Spaces:
Running
Running
n_epoch = 512
Browse files
app.py
CHANGED
@@ -111,7 +111,7 @@ save_dir = './weights/'
|
|
111 |
|
112 |
# training hyperparameters
|
113 |
batch_size = 1000
|
114 |
-
n_epoch =
|
115 |
lrate=1e-3
|
116 |
|
117 |
# construct DDPM noise schedule
|
|
|
111 |
|
112 |
# training hyperparameters
|
113 |
batch_size = 1000
|
114 |
+
n_epoch = 512
|
115 |
lrate=1e-3
|
116 |
|
117 |
# construct DDPM noise schedule
|