higher default learning rate

2026-09-10 12:01:08 +08:00 · 2022-05-16 13:39:55 -07:00
parent cae9f4a71f
commit fc8e4547aa
3 changed files with 3 additions and 3 deletions
@@ -542,7 +542,7 @@ class Trainer(object):
        ema_decay = 0.995,
        image_size = 128,
        train_batch_size = 32,
-        train_lr = 2e-5,
+        train_lr = 1e-4,
        train_num_steps = 100000,
        gradient_accumulate_every = 2,
        amp = False,