summaryrefslogtreecommitdiffstats
path: root/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py
diff options
context:
space:
mode:
Diffstat (limited to 'model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py')
-rwxr-xr-xmodel-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py38
1 files changed, 0 insertions, 38 deletions
diff --git a/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py b/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py
deleted file mode 100755
index 1296d84e180..00000000000
--- a/model-integration/src/test/models/onnx/pytorch/pytorch_one_layer.py
+++ /dev/null
@@ -1,38 +0,0 @@
-# Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
-
-import torch
-import torch.onnx
-
-
-class MyModel(torch.nn.Module):
- def __init__(self):
- super(MyModel, self).__init__()
- self.linear = torch.nn.Linear(in_features=3, out_features=1)
- self.logistic = torch.nn.Sigmoid()
-
- def forward(self, vec):
- return self.logistic(self.linear(vec))
-
-
-def main():
- model = MyModel()
-
- # Omit training - just export randomly initialized network
-
- data = torch.FloatTensor([[0.1, 0.2, 0.3],[0.4, 0.5, 0.6]])
- torch.onnx.export(model,
- data,
- "one_layer.onnx",
- input_names = ["input"],
- output_names = ["output"],
- dynamic_axes = {
- "input": {0: "batch"},
- "output": {0: "batch"},
- },
- opset_version=12)
-
-
-if __name__ == "__main__":
- main()
-
-