| 80 | |
| 81 | def test_dynamic_batch_size(): |
| 82 | class SimpleModel(torch.nn.Module): |
| 83 | def __init__(self): |
| 84 | super(SimpleModel, self).__init__() |
| 85 | |
| 86 | def forward(self, x): |
| 87 | return x + 2 |
| 88 | |
| 89 | net = SimpleModel() |
| 90 | dummy_input = torch.randn(2, 3, 4, 5) |
no outgoing calls