Allow modifying run_examples.sh defaults (#8769)

This commit is contained in:
Carlos Mocholí 2021-08-09 13:54:35 +02:00 committed by GitHub
parent 87093a3339
commit e7440d68c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions

View File

@ -4,6 +4,6 @@ set -ex
dir_path=$(dirname "${BASH_SOURCE[0]}")
args="--trainer.max_epochs=1 --data.batch_size=32 --trainer.limit_train_batches=2 --trainer.limit_val_batches=2"
python "${dir_path}/basic_examples/simple_image_classifier.py" "$@" ${args}
python "${dir_path}/basic_examples/backbone_image_classifier.py" "$@" ${args}
python "${dir_path}/basic_examples/autoencoder.py" "$@" ${args}
python "${dir_path}/basic_examples/simple_image_classifier.py" ${args} "$@"
python "${dir_path}/basic_examples/backbone_image_classifier.py" ${args} "$@"
python "${dir_path}/basic_examples/autoencoder.py" ${args} "$@"