From 4f66fde849be6a282756b94078af4c47fb03a196 Mon Sep 17 00:00:00 2001 From: "Florian C.F. Schulz" Date: Tue, 25 May 2021 13:53:38 +0200 Subject: [PATCH] Update swav_aug.py --- augmentations/swav_aug.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)