Simplify code, delete redundancy line
delete redundancy line 597 `if args.train` which is the same function to line 547, in order to simplify code.
wouldn't this cause some kind of indentation error? (I don't have time to test the change sorry)