更新 'train_for_c2net.py'
This commit is contained in:
parent
864fce7e21
commit
3e0fb36906
|
@ -39,7 +39,7 @@ parser.add_argument('--epoch_size',
|
||||||
set_seed(1)
|
set_seed(1)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
args = parser.parse_args()
|
args, unknown = parser.parse_known_args()
|
||||||
print('args:')
|
print('args:')
|
||||||
print(args)
|
print(args)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue