Issue with Calculating Per-Item Losses using Segmentation Models

This repository seems to already have a fastai loss wrapper. Have a look if you like:

2 Likes