Does Pearl support serving in TorchScript?

facebookresearch / Pearl

A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.

MIT License

2.6k stars 157 forks source link

Does Pearl support serving in TorchScript? #56

Closed ValerieF412 closed 8 months ago

ValerieF412 commented 8 months ago

Hi, can I ask if Pearl currently support using torch.jit.script to serialize the model in TorchScript?

xuruiyang commented 8 months ago

Yes, Pearl uses PyTorch nn.Module to build models, so it should be able to be serialized by torch.jit.script. Please let us know if you get any issues.