no update when traiing is false

examples/test_all.sh

@@ -19,7 +19,7 @@ memory_type=("historical")
 #memory_type=("local" "all_update" "historical" "all_reduce")
 shared_memory_ssim=("0.3")
 #data_param=("WIKI" "REDDIT" "LASTFM" "WikiTalk")
-data_param=("WikiTalk")
+data_param=("LASTFM" "WikiTalk" "StackOverflow" "GDELT")
 # "StackOverflow" "GDELT") 
 #"GDELT")
 #data_param=("WIKI" "REDDIT" "LASTFM" "DGraphFin" "WikiTalk" "StackOverflow")
@@ -33,9 +33,9 @@ data_param=("WikiTalk")
 #seed=(( RANDOM % 1000000 + 1 ))
 mkdir -p all_"$seed"
 for data in "${data_param[@]}"; do
-    model="JODIE_large"
+    model="TGN_large"
     if [ "$data" = "WIKI" ] || [ "$data" = "REDDIT" ] || [ "$data" = "LASTFM" ]; then
-        model="JODIE"
+        model="TGN"
     fi
     #model="APAN"
     mkdir all_"$seed"/"$data"