Trying to train an object detection model on the Citypersons dataset following this tutorial. https://neptune.ai/blog/how-to-train-your-own-object-detector-using-tensorflow-object-detection-api
I run the following command:
python model_main_tf2.py --pipeline_config_path=models/MaskCNN/v1/pipeline.config -- model_dir=models/MaskCNN/v1/ --checkpoint_every_n=4 --num_workers=2 --alsologtostderr
And getting the following error:
File "/Users/Desktop/TensorFlow/tf2_api_env/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 7215, in raise_from_not_ok_status
raise core._status_to_exception(e) from None # pylint: disable=protected-access
tensorflow.python.framework.errors_impl.InvalidArgumentError: {{function_node __wrapped__IteratorGetNext_output_types_18_device_/job:localhost/replica:0/task:0/device:CPU:0}} indices[0] = 0 is not in [0, 0)
[[{{node GatherV2_7}}]]
[[MultiDeviceIteratorGetNextFromShard]]
[[RemoteCall]] [Op:IteratorGetNext]
I am fairly new to this and can't figure out the problem, any help would be greatly appreciated. Thanks.
