Learner.export() pickling fails with FocalLossFlat


Hi everyone,
how can I pass the FocalLoss to the constructor such that it can be pickled when .export( )?

Thank you!

This is a known error with an open GitHub issue. I don’t think there’s a workaround other than switching loss functions to something that is pickleable and switching back to focal loss after loading.

1 Like