search

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

多gpu训练报错 #17

Open cl886699 opened 4 years ago

cl886699 commented 4 years ago

你好，我用多gpu训练模型时，报了以下错误，我想要多gpu训练，还需要更改哪些呢 faster_rcnn\detection\utils\misc.py:53 calc_batch_padded_shape * return tf.cast(tf.reduce_max(meta[:, 6:8], axis=0), tf.int32).numpy()

AttributeError: 'Tensor' object has no attribute 'numpy'
cl886699 commented 4 years ago

我改了一个多gup训练的版本，但只支持固定大小的输入