cnn_traced = torch.jit.trace(cnn_model, torch.rand([1,3,224,224])) ... in this small program, though most of it should remind you of the Python PyTorch API.
確定! 回上一頁