fix ConsinFixmatch bug

2022-10-25 03:41:39 +00:00 · 2022-10-25 03:41:39 +00:00 · 3be87e0d91
parent 944763d7a5
commit 3be87e0d91
1 changed files with 2 additions and 1 deletions
--- a/ppcls/optimizer/learning_rate.py
+++ b/ppcls/optimizer/learning_rate.py
@ -502,6 +502,7 @@ class CosineFixmatch(LRBase):
        self.num_warmup_steps = num_warmup_steps
        self.num_cycles = num_cycles
        self.last_epoch = last_epoch
+        self.by_epoch = by_epoch

    def __call__(self):
        def _lr_lambda(current_step):