Error while using tlt-infer with mask_rcnn on a custom dataset

[pejc]

Alright, it is the solution. It wasn’t specified in the TLT docs, so I wasn’t sure.

But now I get the same error as this guy : Error running MaskRCNN inference after custom training - #6 by hyperlight

Any ETA on the fix?