Skip to content

GitLab

Unverified Commit 2fbf725f authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Fix NAS unittest (#3559)

parent 78fc0624
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
test/scripts/nas.sh
View file @ 2fbf725f
...@@ -7,7 +7,7 @@ echo "===========================Testing: NAS===========================" ...@@ -7,7 +7,7 @@ echo "===========================Testing: NAS==========================="
EXAMPLE_DIR=${CWD}/../examples/nas EXAMPLE_DIR=${CWD}/../examples/nas
echo "testing nnictl ss_gen (classic nas)..." echo "testing nnictl ss_gen (classic nas)..."
cd $EXAMPLE_DIR/classic_nas cd $EXAMPLE_DIR/legacy/classic_nas
SEARCH_SPACE_JSON=nni_auto_gen_search_space.json SEARCH_SPACE_JSON=nni_auto_gen_search_space.json
if [ -f $SEARCH_SPACE_JSON ]; then if [ -f $SEARCH_SPACE_JSON ]; then
rm $SEARCH_SPACE_JSON rm $SEARCH_SPACE_JSON
...@@ -19,12 +19,12 @@ if [ ! -f $SEARCH_SPACE_JSON ]; then ...@@ -19,12 +19,12 @@ if [ ! -f $SEARCH_SPACE_JSON ]; then
fi fi
echo "testing darts..." echo "testing darts..."
cd $EXAMPLE_DIR/darts cd $EXAMPLE_DIR/oneshot/darts
python3 search.py --epochs 1 --channels 2 --layers 4 python3 search.py --epochs 1 --channels 2 --layers 4
python3 retrain.py --arc-checkpoint ./checkpoint.json --layers 4 --epochs 1 python3 retrain.py --arc-checkpoint ./checkpoint.json --layers 4 --epochs 1
echo "testing enas..." echo "testing enas..."
cd $EXAMPLE_DIR/enas cd $EXAMPLE_DIR/oneshot/enas
python3 search.py --search-for macro --epochs 1 python3 search.py --search-for macro --epochs 1
python3 search.py --search-for micro --epochs 1 python3 search.py --search-for micro --epochs 1
...@@ -34,5 +34,5 @@ python3 search.py --search-for micro --epochs 1 ...@@ -34,5 +34,5 @@ python3 search.py --search-for micro --epochs 1
#python3 train.py #python3 train.py
echo "testing pdarts..." echo "testing pdarts..."
cd $EXAMPLE_DIR/pdarts cd $EXAMPLE_DIR/legacy/pdarts
python3 search.py --epochs 1 --channels 4 --nodes 2 --log-frequency 10 --add_layers 0 --add_layers 1 --dropped_ops 3 --dropped_ops 3 python3 search.py --epochs 1 --channels 4 --nodes 2 --log-frequency 10 --add_layers 0 --add_layers 1 --dropped_ops 3 --dropped_ops 3
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment