Error running train.py
ItsCRC opened this issue · 0 comments
ItsCRC commented
When I run the command: python train.py --model_dir MODEL_DIR --pre_trained_model PRE_TRAINED_MODEL, where MODEL_DIR=model and PRE_TRAINED_MODEL=pre_trained_model. The error pops out and says: Couldn't find the checkpoint file or checkpoints in the given directory 'pre_trained_model'.
I have checkpoints in 'model' dir and resnet in 'pre_trained_model' as said in the setup section. How to solve this????