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