Skip to content

Commit a73ae3e

Browse files
committed
Better default for AdamW
1 parent 06505ba commit a73ae3e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/train_unconditional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ def transforms(examples):
195195
parser.add_argument("--adam_beta1", type=float, default=0.95)
196196
parser.add_argument("--adam_beta2", type=float, default=0.999)
197197
parser.add_argument("--adam_weight_decay", type=float, default=1e-6)
198-
parser.add_argument("--adam_epsilon", type=float, default=1e-3)
198+
parser.add_argument("--adam_epsilon", type=float, default=1e-08)
199199
parser.add_argument("--use_ema", action="store_true", default=True)
200200
parser.add_argument("--ema_inv_gamma", type=float, default=1.0)
201201
parser.add_argument("--ema_power", type=float, default=3 / 4)

0 commit comments

Comments
 (0)