Hello @ptrblck, Could you please help me with above problem. I would really appreciate it if you have a look at this problem. Thank you so much.
My code kinda works well for 1 class (or should say 2 classes with background) but it crashes when trained on multiple classes + background.