We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 06505ba commit a73ae3eCopy full SHA for a73ae3e
examples/train_unconditional.py
@@ -195,7 +195,7 @@ def transforms(examples):
195
parser.add_argument("--adam_beta1", type=float, default=0.95)
196
parser.add_argument("--adam_beta2", type=float, default=0.999)
197
parser.add_argument("--adam_weight_decay", type=float, default=1e-6)
198
- parser.add_argument("--adam_epsilon", type=float, default=1e-3)
+ parser.add_argument("--adam_epsilon", type=float, default=1e-08)
199
parser.add_argument("--use_ema", action="store_true", default=True)
200
parser.add_argument("--ema_inv_gamma", type=float, default=1.0)
201
parser.add_argument("--ema_power", type=float, default=3 / 4)
0 commit comments