Update README.md

daiquocnguyen · Aug 16, 2022 · 99c88a1 · 99c88a1
1 parent 7dc290c
commit 99c88a1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -46,7 +46,9 @@ Variant 1            |  Variant 2
 
 - Applying an unsupervised transductive setting for graph classification: 
 
-		$ python train_UGformerV1_UnSup.py --dataset PTC --batch_size 2 --degree_as_tag --ff_hidden_size 1024 --num_neighbors 4 --num_sampled 512 --num_epochs 50 --num_timesteps 2 --learning_rate 0.0001 --model_name PTC_bs2_dro05_1024_4_idx0_2_2
+		Tensorflow: $ python train_UGformerV1_UnSup.py --dataset PTC --batch_size 2 --degree_as_tag --ff_hidden_size 1024 --num_neighbors 4 --num_sampled 512 --num_epochs 50 --num_timesteps 2 --learning_rate 0.0001 --model_name PTC_bs2_dro05_1024_4_idx0_2_2
+		
+		PyTorch: $ python train_UGformerV1_UnSup.py --dataset PTC --batch_size 2 --ff_hidden_size 1024 --num_neighbors 4 --sampled_num 512 --num_epochs 50 --num_timesteps 2 --learning_rate 0.0001 --model_name PTC_bs2_dro05_1024_4_idx0_2_2
 
 
 #### Requirements