thanks for your replay,
how can i exported model to torch script and than load it back and export to onnx?
to save as torch script i am using the following code:
model.eval()
dummy_input = torch.randn(1, 3, 224, 224)
traced = torch.jit.trace(model, dummy_input)
traced.save("/app_data/test_torch_script/torch_script_test.zip")

can i change the above code in order to use it’s output for exporting to onnx?