Model is taking more than 1 hour to complete single epoch even with colab GPU

[saikiran159]

Hi, i just cloned your code (used your dog images for training) and runned in colab(GPU) (changed keepdims to keep_dims in model.py file in mrccn folder). You told this took about half an hour to complete the training(in hackernoon blog). but for me it is taking more than 1 hour to complete a single epoch.
Installed
1 tensorflow==1.3.0
2. keras==2.0.8

I downloaded mask_rcnn_coco.h5 from here. [https://github.com/matterport/Mask_RCNN/releases/download/v2.0/mask_rcnn_coco.h5]

It means a lot if you solved this issue for me. Thank u.

[nicolas-lair]

Use tensorflow-gpu==1.3.0 to effectively use GPU.