pip install requirements.txt

python pretrain.py --config=pretrain.config.json --model=resnet18 --batch_size=512 --pred_dim=512 --encoder_dim=2048 --num_epochs=800 --device=gpu --num_workers=1

python linear_classifier.py --config=lcls.config.json --num_epochs=100 --batch_size=256 --device=gpu

python linear_classifier.py --config=lcls.config.json --num_epochs=100 --batch_size=128 --device=gpu --base_line_eval=True