search

rishizek / tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow
MIT License
286 stars 102 forks source link

Error running train.py #40

Open ItsCRC opened 3 years ago

ItsCRC commented 3 years ago

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????