Setting the number of epochs to 2 just to test quickly
@@ -61,7 +61,7 @@ python3 -m torch.distributed.launch \
--norm_last_layer False \
--use_fp16 False \
--clip_grad 0 \
- --epochs 800 \
+ --epochs 2 \
--global_crops_scale 0.25 1.0 \
--local_crops_number 10 \
--local_crops_scale 0.05 0.25 \
