(self)
| 10 | class TestPyTorch(unittest.TestCase): |
| 11 | # PyTorch smoke test based on http://pytorch.org/tutorials/beginner/nlp/deep_learning_tutorial.html |
| 12 | def test_nn(self): |
| 13 | torch.manual_seed(31337) |
| 14 | linear_torch = tnn.Linear(5,3) |
| 15 | data_torch = autograd.Variable(torch.randn(2, 5)) |
| 16 | linear_torch(data_torch) |
| 17 | |
| 18 | @gpu_test |
| 19 | def test_linalg(self): |
nothing calls this directly
no outgoing calls
no test coverage detected