Giux22/semana4-todos

imagesize = 128 # the generated image resolution trainbatchsize = 16 evalbatchsize = 16 # how many images to sample during evaluation numepochs = 15000 gradientaccumulationsteps = 1 learningrate = 7e-5 lrwarmupsteps = 500 saveimageepochs = 50 savemodel_epochs = 50

preprocess = transforms.Compose( [ transforms.Resize((config.imagesize, config.imagesize)), transforms.RandomRotation(5), # transforms.RandomHorizontalFlip(), # transforms.RandomVerticalFlip(), transforms.ToTensor(), transforms.Normalize([0.5], [0.5]), ] )