Speed_paramters of my own trained models

[bis70]

Hello,

how can I test the speed_paramters of my own trained models. i cant use the pth s

Thank You

[ltkong218]

The running speed and number of parameters do not rely on the trained parameter or state dict. You can load your trained models by model.load_state_dict(torch.load('checkpoints/path/your_model.pth')).

[bis70]

that's right, no idea what I was thinking, thank you