diff --git a/source/pytorch.js b/source/pytorch.js index c38f55d6f2..21b17ac069 100644 --- a/source/pytorch.js +++ b/source/pytorch.js @@ -1978,6 +1978,7 @@ pytorch.Container.Zip.Execution = class extends pytorch.Execution { for (let i = 0; i < schema.outputs.length; i++) { const parameter = schema.outputs[i]; switch (parameter.type) { + case 'Scalar': case 'Tensor': { const output = this.invoke('torch.Tensor', []); output.__origin__ = type; diff --git a/test/models.json b/test/models.json index ac5f479dda..ff708ee161 100644 --- a/test/models.json +++ b/test/models.json @@ -4362,6 +4362,14 @@ "format": "TorchScript v1.3", "link": "https://github.com/lutzroeder/netron/issues/281" }, + { + "type": "pytorch", + "target": "mask_model.pt", + "source": "https://github.com/lutzroeder/netron/files/10080302/mask_model.pt.zip[mask_model.pt]", + "format": "TorchScript v1.7", + "error": "Unsupported torch.add expression type in 'mask_model.pt'.", + "link": "https://github.com/lutzroeder/netron/issues/842" + }, { "type": "pytorch", "target": "mask_r_cnn.pth",