pytorch onnx export error
![[Pytorch] ONNX export할 때 RuntimeError: step!=1 is currently not supported 에러 해결 방법](https://img1.daumcdn.net/thumb/R750x0/?scode=mtistory2&fname=https%3A%2F%2Fblog.kakaocdn.net%2Fdn%2FbVTUe4%2FbtsuqBKtkwm%2FXbMe5gNKcV6kqX2puXeaE0%2Fimg.png)
[Pytorch] ONNX export할 때 RuntimeError: step!=1 is currently not supported 에러 해결 방법
torch.onnx.export할 때 RuntimeError: step!=1 is currently not supported 에러 해결 방법 서론 아래 코드와 같이 모델을 onnx로 변환하고자 했는데 import torch import torch.onnx model = get_net(False) params = torch.load("/home/slrnd/test/00.work/BDD/YOLOP/runs/BddDataset/_2023-09-15-11-11/final_state.pth", map_location="cpu") model.load_state_dict(params) model.cpu() params = model.state_dict() torch.save(params, "./net.prm", pi..