updated epoch setting

phy-q · Dec 25, 2021 · e68730d · e68730d
1 parent 6658d04
commit e68730d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sciencebirdsagents/PhyreAgents/RPIN/train.py b/sciencebirdsagents/PhyreAgents/RPIN/train.py
@@ -137,7 +137,7 @@ def main():
     )
     print(f'size: train {len(train_loader)} / test {len(val_loader)}')
 
-    cfg.SOLVER.MAX_ITERS = len(train_loader) * 5 * cfg.SOLVER.BATCH_SIZE if args.protocal == 'template' else len(train_loader) * 10 * cfg.SOLVER.BATCH_SIZE
+    cfg.SOLVER.MAX_ITERS = len(train_loader) * 5 * cfg.SOLVER.BATCH_SIZE
     cfg.SOLVER.VAL_INTERVAL = len(train_loader) // 1 * cfg.SOLVER.BATCH_SIZE
     print(cfg.SOLVER.MAX_ITERS)
     print(cfg.SOLVER.VAL_INTERVAL)