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

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

Open saikiran159 opened this issue 4 years ago • 1 comments

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.

saikiran159 avatar Aug 29 '20 13:08 saikiran159

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

nicolas-lair avatar Jan 17 '22 22:01 nicolas-lair