コード例 #1
0
 def test_TinyCNN(self):
     model = models.TinyCNN(in_channels=3, out_channels=3)
     self._smoke_test(model, in_channels=3)
コード例 #2
0
 def test_TinyCNN(self):
   model = models.TinyCNN(in_channels=1)
   self._smoke_test(model)