I am trying to train MaskRCNN using TLT. The training with the COCO data set was successful. Training with a custom data set was fine and the training ended with ‘success’ status. The evaluation was also fine. At the inference stage an error message appears saying “The batch size should not be larger than the number of samples”, and then the process run failed.
I will appreciate your help