To use the standardized environment on GPU 0 run:
./docker_build # this may take several minutes
./docker_run -g 0
Download example checkpoints
wget --no-check-certificate 'https://drive.google.com/u/0/uc?export=download&confirm=1wnBBQYG21b_rvGlmDi9kboTxcDlbRV2K' -O checkpoints.zip
https://drive.google.com/file/d/1wnBBQYG21b_rvGlmDi9kboTxcDlbRV2K/view?usp=sharing
Run Training and Latency Benchmarks
python train.py --model configs/model/gain_bottleneck_1.0.yaml --clean --dataset configs/dataset/pets.yaml --project test --wandb
while true; do python eval.py --model configs/model/gain_bottleneck_1.0.yaml --dataset configs/dataset/pets.yaml --project test; done