更新 'train_for_c2net.py'

2022-07-13 15:21:46 +08:00 · 2022-07-13 15:21:46 +08:00 · 3e0fb36906
parent 864fce7e21
commit 3e0fb36906
1 changed files with 1 additions and 1 deletions
--- a/train_for_c2net.py
+++ b/train_for_c2net.py
@ -39,7 +39,7 @@ parser.add_argument('--epoch_size',
 set_seed(1)

 if __name__ == "__main__":
-    args = parser.parse_args()
+    args, unknown = parser.parse_known_args()
    print('args:')
    print(args)