Update prepare.sh

hoszeman · Jun 10, 2022 · 0098727 · 0098727
1 parent ff2c144
commit 0098727
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
@@ -394,6 +394,7 @@ elif [ ${model_name} == "fgcnn" ]; then
         cd ../..
         cp -r ./models/rank/fgcnn/data/trainlite/* ./test_tipc/data/train
         cp -r ./datasets/criteo_fgcnn/test/valid.h5 ./test_tipc/data/infer
+    fi
 elif [ ${model_name} == "iprec" ]; then
     mkdir -p ./test_tipc/data/train
     mkdir -p ./test_tipc/data/infer