Viredery / tf-eager-fasterrcnn

Faster R-CNN R-101-FPN model was implemented with TensorFlow2.0 eager execution.
MIT License
89 stars 46 forks source link

InvalidArgumentError: indices[204] = [204, -1] does not index into shape [256,81,4] [Op:TensorScatterUpdate] #15

Closed arunraja-hub closed 4 years ago

arunraja-hub commented 4 years ago

I tried running train_model.ipynb. My tf version is 2.0.0 and cuda version is 10.0. But I am still getting the following error: image Could you help me rectify this?

Viredery commented 4 years ago

Please follow this comment and it may be helpful for you https://github.com/Viredery/tf-eager-fasterrcnn/issues/10#issuecomment-629824408