search

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
https://sthalles.github.io/simple-self-supervised-learning/
MIT License
2.19k stars 457 forks source link

keyword arguments to the run.py file #36

Open sourabhsugandhi opened 3 years ago

sourabhsugandhi commented 3 years ago

for smooth execution use run.py use following command

BEFORE : $ python run.py -data ./datasets --dataset-name stl10 --log-every-n-steps 100 --epochs 100 AFTER : $ python run.py -data ./datasets -dataset-name stl10 --log-every-n-steps 100 --epochs 100

Or else you can make change in run.py at line number 16 from parser.add_argument('-dataset-name', default='stl10',help='dataset name', choices=['stl10', 'cifar10']) to parser.add_argument('--dataset-name', default='stl10', help='dataset name', choices=['stl10', 'cifar10'])