提交 3be87e0d 编写于 作者: D dongshuilong 提交者: Walter

fix ConsinFixmatch bug

上级 944763d7
......@@ -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
\ No newline at end of file
return learning_rate
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册