torch.onnx.export expects the model as the first argument, while it seems you are passing the state_dict to it,
Could you create an instance of the model, load the state_dict via model.load_state_dict(state_dict), and try to export it afterwards?