train.py line 6 is from arg_parser import *, but there is no arg_parser.py in the repository. Running the documented training command therefore fails at import:
Traceback (most recent call last):
File "train.py", line 6, in <module>
from arg_parser import *
ModuleNotFoundError: No module named 'arg_parser'
It looks like arg_parser.py was omitted from the upload, and without it the training workflow cannot be run as shipped.
Would you be able to add arg_parser.py to the repository when you get a chance? Thanks very much for sharing the code.
train.py line 6 is
from arg_parser import *, but there is no arg_parser.py in the repository. Running the documented training command therefore fails at import:It looks like arg_parser.py was omitted from the upload, and without it the training workflow cannot be run as shipped.
Would you be able to add arg_parser.py to the repository when you get a chance? Thanks very much for sharing the code.