Please follow the following instructions to install the required packages:
cd code;
python -m venv venv
source venv/bin/activate
python3 -m pip install --upgrade pip
pip install -r requirements.txt
Or alternatively, run the setup.sh
script.
you can train the model using the following command:
cd code; python -m train --config=configs/fashion-mnist-gfz.py --config.checkpoint_name="checkpoint-name" --config.num_epochs=1
the params checkpoints after each epoch will be stored under checkpoints
. there is a load_checkpoint
function to load them back.
To compute evaluation metrics on a given checkpoint, use the following command:
cd code; python -m evaluate --checkpoint=dfz-2-epochs-first-try-1