Пример #1
0
 def test_MockModel(self):
     mp = MockPosterior()
     mm = MockModel(mp)
     X = torch.empty(0)
     self.assertEqual(mm.posterior(X), mp)
     self.assertEqual(mm.num_outputs, 0)
     mm.state_dict()
     mm.load_state_dict()