Update and rename run_KT.sh to train_KT.sh

a4152684 · Dec 1, 2022 · c5add2c · c5add2c
1 parent 0e96dbe
commit c5add2c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/run_KT.sh → train_KT.sh b/run_KT.sh → train_KT.sh
@@ -4,11 +4,10 @@ epochs=600
 save_frequency=200
 lr=1e-4
 batch_size=32
-dataset_name=3depn
-# dataset_name=crn
+# dataset_name=3depn
+dataset_name=crn
 
 for category in plane #cabinet car chair lamp sofa table watercraft
 do
     sh run/train-KT.sh ${category} ${epochs} ${save_frequency} ${latent_dim} ${lr} ${dataset_name} ${batch_size}
-    # sh run/test-KT.sh ${category} ${latent_dim} ${dataset_name}
 done