jsiloto / split_mobilenetv3

concept-drift-split-models

Environment

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