* fix typos

no functional change

* Update hpo.py

better fix for typo

no functional change
This commit is contained in:
RainRat authored and GitHub committed 2024-05-29 22:09:30 +02:00
1 parent 892e8a8263
commit 2b7bcf664b
4 files changed
+4 -4

No files matched your search

+1 -1
View File
@@ -353,7 +353,7 @@ def classify_albumentations(
if vflip > 0:
T += [A.VerticalFlip(p=vflip)]
if jitter > 0:
color_jitter = (float(jitter),) * 3 # repeat value for brightness, contrast, satuaration, 0 hue
color_jitter = (float(jitter),) * 3 # repeat value for brightness, contrast, saturation, 0 hue
T += [A.ColorJitter(*color_jitter, 0)]
else: # Use fixed crop for eval set (reproducibility)
T = [A.SmallestMaxSize(max_size=size), A.CenterCrop(height=size, width=size)]