python3 run_script.py --cosine-anneal --filename original_output.pth --base-lr 1e-3 --test-freq 2 \
--run-name test_run \
--lambda_ 0.01 \
--num-epochs 1000 \
--batch-size 256 \
--temperature 0.05 \
--temp_z 0.1 \
--no_color_distor \
--dataset colorMNIST \
--arch LeNet \
--save_freq 100 \
--projection-mode 'linear' \
--z-init 'normal' \
--lsh 'default' \
--lsh-adjacency 'adjacent' \
--attention-weightdecay 0.1 \