diff --git a/classify/train.py b/classify/train.py index 6f9eda40be62..5ae6980716a4 100644 --- a/classify/train.py +++ b/classify/train.py @@ -177,6 +177,7 @@ def train(opt, device): # Scheduler lrf = 0.01 # final lr (fraction of lr0) + # lf = lambda x: ((1 + math.cos(x * math.pi / epochs)) / 2) * (1 - lrf) + lrf # cosine def lf(x): return (1 - x / epochs) * (1 - lrf) + lrf # linear diff --git a/requirements.txt b/requirements.txt index e496ad8206b6..3b2674596258 100644 --- a/requirements.txt +++ b/requirements.txt @@ -47,4 +47,3 @@ setuptools>=65.5.1 # Snyk vulnerability fix # mss # screenshots # albumentations>=1.0.3 # pycocotools>=2.0.6 # COCO mAP -wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability