Add autoencoder script to train_all_models script

analogdevicesinc · Feb 5, 2024 · f54de41 · f54de41
1 parent 4637759
commit f54de41
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/train_all_models.sh b/train_all_models.sh
@@ -59,3 +59,6 @@ scripts/train_bayer2rgb_imagenet.sh "$@"
 echo "-----------------------------"
 echo "Training kws20_nas model"
 scripts/train_kws20_nas.sh "$@"
+echo "-----------------------------"
+echo "Training Auto Encoder model"
+scripts/train_autoencoder.sh "$@"