Skip to content

Commit aa332b5

Browse files
committed
Add missing files
1 parent cb70069 commit aa332b5

File tree

3 files changed

+23
-0
lines changed

3 files changed

+23
-0
lines changed
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:7be546e4ea8636bd79c26e4ff06e28a70a4f8082f8b9c28fbcae6adb0e46e24d
3+
size 143
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:d58ef7dc37681521fa8c8856f4c256b614b535efdc33e838836fee3bde9106da
3+
size 1317
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
import torch
2+
3+
class MyModule(torch.nn.Module):
4+
def __init__(self):
5+
super(MyModule, self).__init__()
6+
7+
def forward(self, a, b):
8+
return a, torch.tensor([])
9+
10+
11+
my_module = MyModule()
12+
13+
dummy = (torch.rand(2), torch.rand(2))
14+
torch.onnx.export(my_module, dummy, "batchdim_mismatch.onnx")
15+
16+
my_module_traced = torch.jit.trace(my_module, dummy)
17+
torch.jit.save(my_module_traced, "batchdim_mismatch.pt")

0 commit comments

Comments
 (0)