from params import Params
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--exp_name", type=str, default="baseline")
parser.add_argument("--model", type=str, default="resnet18")
parser.add_argument("--dataset", type=str, default="cifar10")
args = parser.parse_args()
params = Params()
params.exp_name = args.exp_name
params.model = args.model
params.dataset = args.dataset
params.build()
print(params)
torch.cuda.manual_seed_all(params.seed)
# =======================================
# Data
print('==> Preparing dataset %s' % params.dataset)
transform_train = transforms.Compose([
transforms.RandomCrop(32, padding=4),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize((0.4914, 0.4822, 0.4465), (0.2023, 0.1994, 0.2010)),
])
transform_test = transforms.Compose([
