若要匯出模型,您將使用 torch.onnx.export() 函數。 ... mode model.eval() # Let's create a dummy input tensor dummy_input = torch.randn(1, ...
確定! 回上一頁