sam125

Is happening the same here. I get stuck on the first epoch. Checkpoint Path: /content/Mask_RCNN_master_Clean_Colab/logs/object20230526T2334/mask_rcnn_object_{epoch:04d}.h5 Selecting layers to train fpn_c5p5 (Conv2D) fpn_c4p4 (Conv2D) fpn_c3p3 (Conv2D) fpn_c2p2 (Conv2D) fpn_p5 (Conv2D) fpn_p2...

I think tensorflow is only using the CPU insted of the GPU