fix arugment parsing part in train.py #24

Closed jjkim0807 closed 7 months ago