diff --git a/augmentations/swav_aug.py b/augmentations/swav_aug.py index 274ca53..f0ca361 100644 --- a/augmentations/swav_aug.py +++ b/augmentations/swav_aug.py @@ -9,7 +9,7 @@ import numpy as np import torchvision.transforms as transforms -class SvAVTransform(): +class SwAVTransform(): def __init__(self, size_crops, nmb_crops, min_scale_crops, max_scale_crops): assert len(size_crops) == len(nmb_crops) assert len(min_scale_crops) == len(nmb_crops)