Skip to content

Commit

Permalink
Fix typos
Browse files Browse the repository at this point in the history
  • Loading branch information
IgorSusmelj committed Jul 17, 2023
1 parent a07223e commit e39b321
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion benchmarks/imagenet/resnet50/byol.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,6 +145,6 @@ def configure_optimizers(self):


# BYOL uses a slight modification of the SimCLR transforms.
# Iuses asymmetric augmentation and solarize.
# It uses asymmetric augmentation and solarize.
# Check table 6 in the BYOL paper for more info.
transform = BYOLTransform()
Original file line number Diff line number Diff line change
Expand Up @@ -239,7 +239,7 @@ def create_dataset_train_ssl(model):
"""
model_to_transform = {
BarlowTwinsModel: byol_transform,
BYOLModel: byol_transform, # for cifar10 we use SimCLR transforms
BYOLModel: byol_transform,
DCL: simclr_transform,
DCLW: simclr_transform,
DINOModel: dino_transform,
Expand Down

0 comments on commit e39b321

Please sign in to comment.