diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 378d241ba9d27538ebc2f79466d7d82baff9f080..e468b29a356fa9e47d4b4cd353a1c30fe69c9729 100755 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -141,6 +141,11 @@ class Engine(object): # set max iteration per epoch mannualy, when training by iteration(s), such as XBM, FixMatch. self.iter_per_epoch = self.config["Global"].get( "iter_per_epoch") + if self.iter_per_epoch < self.update_freq: + logger.warning( + "The arg Global.update_freq greater than iter_per_epoch and has been set to 1. This may be caused by too few of batches." + ) + self.update_freq = 1 self.iter_per_epoch = self.iter_per_epoch // self.update_freq * self.update_freq if self.mode == "eval" or (self.mode == "train" and