It seems you are loading the state_dict into the torch_model, which is containing the parameters and buffers.
To properly load a PyTorch model, you would have to create the model object first and load the state_dict afterwards as described here 17.