https://flyte.org logo
#ask-the-community
Title
# ask-the-community
g

Geert

08/09/2023, 1:55 PM
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!
k

Ketan (kumare3)

08/09/2023, 2:51 PM
You will need to add a transformer For it
It will default to pickle