And finally, the training is complete and “[INFO] Training finished successfully” is displayed. But Evaluation Metrics are all 0, and can not achieve right mask on inferences.
• Hardware : Running TAO Toolkit on Google Colab
• Network Type : Mask_rcnn
• Training spec file : maskrcnn_train_resnet50.txt (2.1 KB)
The reason why I set it to 10, is because I just use 20 training pictures to try to run the demo successfully first, afterwards, I will enlarge the amount of training pictures.
So do you think I still need to enlarge total_steps to this kind of big?
There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks