64

app.py

@@ -111,7 +111,7 @@ save_dir = './weights/'
 
 # training hyperparameters
 batch_size = 100
-n_epoch = 32
+n_epoch = 64
 lrate=1e-3
 
 # construct DDPM noise schedule