update

zchoi · Jun 23, 2022 · be0a8b7 · be0a8b7
1 parent a7fb710
commit be0a8b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_iterative.py b/train_iterative.py
@@ -200,7 +200,7 @@ def train(cfg):
     logging.info("Start training........")
     for epoch in range(start_epoch, cfg.train.max_epochs):
         logging.info('>>>>>> epoch {} <<<<<<'.format(epoch))
-        if epoch < 0:
+        if epoch < 5:
             retrieve_model.train()
             count = 0
             batch_mse_sum = 0.0