This message was deleted.

are the loss function and metrics in a separate module? if yes, you may want to include it in

custom_objects

when do you

save_model()

i.e

bentoml.transformers.save_model(custom_objects=[a,b])

otherwise, would be helpful if you could provide us a minimal reproducible example!

oh I see. Thanks Jian, custom_objects worked for me