maskrcnn_train_tensorflow_colab icon indicating copy to clipboard operation
maskrcnn_train_tensorflow_colab copied to clipboard

Published 20 hours ago • verified publisher icon RomRoc

Can't train model in colab

Open lucydjo opened this issue 3 years ago • 1 comments

Hello !

Step : Train Model

%cd ~/Mask_RCNN
!python dog.py train --dataset=dataset/ --weights=coco

TypeError: Could not build a TypeSpec for <KerasTensor: shape=(None, None, 4) dtype=float32 (created by layer 'tf.math.truediv')> with type KerasTensor

How to fix ?

lucydjo avatar Mar 19 '21 10:03 lucydjo

Hello,

Try with %tensorflow_version 1.x at the beginning of Download and extract dataset section.

gsg213 avatar Apr 09 '21 17:04 gsg213