修正前一次提交备注:修改数据加载器逻辑，保留多类边

farkguidao · Jan 25, 2022 · 1aac78a · 1aac78a
1 parent 2aed881
commit 1aac78a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -47,7 +47,7 @@ def test(parser):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--setting_path',type=str,default='settings/ama_settings.yaml')
+    parser.add_argument('--setting_path',type=str,default='settings/yot_settings.yaml')
     parser.add_argument("--test", action='store_true', help='test or train')
     temp_args, _ = parser.parse_known_args()
     if temp_args.test: