Update main.py

leaderj1001 · Nov 24, 2020 · 3a51b6a · 3a51b6a
1 parent 189f89d
commit 3a51b6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -94,7 +94,7 @@ def main(args):
 
     if not args.evaluation:
         criterion = nn.CrossEntropyLoss()
-        lr_scheduler = optim.lr_scheduler.CosineAnnealingWarmRestarts(optimizer, T_0=10, T_mult=2, eta_min=0.00001)
+        lr_scheduler = optim.lr_scheduler.CosineAnnealingWarmRestarts(optimizer, T_0=10, T_mult=2, eta_min=0.0001)
 
         global_acc = 0.
         for epoch in range(start_epoch, args.epochs + 1):