Could not use pickle when saving model using task_generator for ScriptModule

[t1endat]

When working with task_genrator for torch.jit.ScriptModule, I found that Model Compression uses pickle to model serialization, but this feature is not supported yet because ScriptModule doesn't have the __getstate__ method. Is there 1 way to avoid this error?

[J-shang]

Hello @tiendatAI ,could you show us a simple case to reproduce your error? do you mean you have torch.jit.ScriptModule part in your model?