mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix #1184, scheduler noise bug during merge madness

Browse Source
This commit is contained in:
Ross Wightman 2022-03-21 13:35:45 -07:00
parent 9440a50c95
commit 7cdd164d77
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/scheduler/scheduler.py
View File

@ -92,6 +92,7 @@ class Scheduler:
def _is_apply_noise(self, t) -> bool:
"""Return True if scheduler in noise range."""
apply_noise = False
if self.noise_range_t is not None:
if isinstance(self.noise_range_t, (list, tuple)):
apply_noise = self.noise_range_t[0] <= t < self.noise_range_t[1]