https://flyte.org
Join Slack
Hi, does Flyte support serializing `torch.ScriptModule` ? (as in/output of Task) Getting: ```Failed ...
# flyte-support
s
Hi, does Flyte support serializing 
torch.ScriptModule
? (as in/output of Task) Getting:
Copy code
Failed to convert outputs of task '{snip}.train_model' at position 0:
  Tried to serialize object __torch__.{snip} which does not have a __getstate__ method defined!
f
You will need to add a transformer For it
It will default to pickle
9 Views
Open in Slack
PreviousNext
Powered by

Flyte

Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Powered by