fix ConsinFixmatch bug

3be87e0d · dongshuilong · Walter · 944763d7 · 3be87e0d
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

ppcls/optimizer/learning_rate.py ppcls/optimizer/learning_rate.py +2 -1

未找到文件。
--- 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):