#!/usr/bin/env bash
export CUDA_VISIBLE_DEVICES=0,1

# Stage 1 training
python model.py --config=efficientnetb0 | tee output_efficientnetb0.txt
python model.py --config=efficientnetb5 | tee output_efficientnetb5.txt
python model.py --config=efficientnetb3 | tee output_efficientnetb3.txt

# These were never run
#python model.py --config=densenet169 | tee output_densenet169.txt
#python model.py --config=seresnext | tee output_seresnext.txt
#python model.py --config=vgg19 | tee output_vgg19.txt

# Stage 2 inference
python model.py --config=efficientnetb0-stage2 | tee output_efficientnetb0-stage2.txt
python model.py --config=efficientnetb5-stage2 | tee output_efficientnetb5-stage2.txt
python model.py --config=efficientnetb3-stage2 | tee output_efficientnetb3-stage2.txt