@akirasosa I tried to follow the instructions in the Repo README, and received the following error:
AttributeError: Can't pickle local object 'dice_loss.<locals>.fn'
Am i doing something incorrect, or is this a common error?