How to perform inference using model weights obtained from m.model.state_dict()?

[cytus-kira]

Hi,

I am able to obtain the network weights using m.model.state_dict() from your project. However, I am unsure about how to calculate the model's predictions. Could you please guide me on how the network performs inference using these parameters? Specifically, I would like to understand how matrix operations are used to compute the final prediction.

Thank you!