Merge pull request #9 from microsoft/mia_adv_attacks

Model save paths made independent of test domain
microsoft · Aug 24, 2020 · 38011b9 · 38011b9
2 parents 10f01fc + 3f2fa45
commit 38011b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -134,7 +134,7 @@
 final_metric_score=[]
 base_res_dir=(
                 "results/" + args.dataset_name + '/' + args.method_name + '/' + args.match_layer 
-                + '/' + 'train_' + str(args.train_domains) + '_test_' + str(args.test_domains) 
+                + '/' + 'train_' + str(args.train_domains)  
             )
 if not os.path.exists(base_res_dir):
     os.makedirs(base_res_dir)    

diff --git a/train.py b/train.py
@@ -116,7 +116,7 @@
 final_report_accuracy=[]
 base_res_dir=(
                 "results/" + args.dataset_name + '/' + args.method_name + '/' + args.match_layer 
-                + '/' + 'train_' + str(args.train_domains) + '_test_' + str(args.test_domains) 
+                + '/' + 'train_' + str(args.train_domains)
             )
 if not os.path.exists(base_res_dir):
     os.makedirs(base_res_dir)