diff --git a/train_all_models.sh b/train_all_models.sh
index cfb4e804f..18fef7f0b 100755
--- 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 "$@"