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):
@ -517,4 +518,4 @@ class CosineFixmatch(LRBase):
            lr_lambda=_lr_lambda,
            last_epoch=self.last_epoch)
        setattr(learning_rate, "by_epoch", self.by_epoch)
-        return learning_rate
+        return learning_rate