batch_size = 500

app.py

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