2021-07-06 12:14:04 +00:00
|
|
|
#!/bin/bash
|
|
|
|
|
set -ex
|
|
|
|
|
|
|
|
|
|
dir_path=$(dirname "${BASH_SOURCE[0]}")
|
2021-08-28 04:43:14 +00:00
|
|
|
args="
|
|
|
|
|
--data.batch_size=32
|
|
|
|
|
--trainer.max_epochs=1
|
|
|
|
|
--trainer.limit_train_batches=2
|
|
|
|
|
--trainer.limit_val_batches=2
|
|
|
|
|
--trainer.limit_test_batches=2
|
|
|
|
|
--trainer.limit_predict_batches=2
|
|
|
|
|
"
|
2021-07-06 12:14:04 +00:00
|
|
|
|
2021-08-09 11:54:35 +00:00
|
|
|
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} "$@"
|
