tf-eager-fasterrcnn icon indicating copy to clipboard operation
tf-eager-fasterrcnn copied to clipboard

多gpu训练报错

Open cl886699 opened this issue 4 years ago • 1 comments

你好,我用多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 avatar Aug 18 '20 09:08 cl886699

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

cl886699 avatar Oct 22 '20 09:10 cl886699