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 aa min_max level clamp

Browse Source
This commit is contained in:
Ross Wightman 2022-12-10 16:43:28 -08:00
parent e3b2f5be0a
commit e7da205345
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/data/auto_augment.py
View File

@ -228,7 +228,7 @@ def _minmax_level_to_arg(level, _hparams, min_val=0., max_val=1.0, clamp=True):
level = (level / _LEVEL_DENOM) level = (level / _LEVEL_DENOM)
min_val + (max_val - min_val) * level min_val + (max_val - min_val) * level
if clamp: if clamp:
level = min(min_val, max(max_val, level)) level = max(min_val, min(max_val, level))
return level, return level,