Solved by Liron_Mor_Yosef in post #3 
              
              
                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_s…