Where does the newly trained model go?

[astralmedia]

I have successfully been able to get training to happen, but I am unsure of where the new model gets saved...I didnt find it anywhere. Thanks so much!

[fletcherist]

hi! it goes to --save-dir

python ${ROOT_DIR}/train.py \
    --dataset-config ${DATASET_CONFIG} \
    --model-config ${MODEL_CONFIG} \
    --name "vae_training" \
    --batch-size ${BATCH_SIZE} \
    --save-dir ${OUTPUT_DIR}