Convert models to torchscript format

[p16i]

Torch provides a way to save models in Torchscript. Using this format, several computations can be further optimised, hence faster models.

More information: https://pytorch.org/docs/stable/jit.html#frequently-asked-questions

We might also to this for datasets (collated_func)

[pritam1309]

can you assign it to me ?

[wannaphong]

can you assign it to me ?

You can work this issues. 👍