barkincavdaroglu / Link-Prediction-Mesh-Network

PyTorch Implementation of a Deep Learning Model for Temporal Link Prediction in MANETs
2 stars 1 forks source link

Torch jit #5

Open barkincavdaroglu opened 1 year ago

barkincavdaroglu commented 1 year ago

Make models compatible with torch jit script so that the model can be run faster.

https://pytorch.org/docs/stable/jit_language_reference.html#language-reference