def test_TinyCNN(self): model = models.TinyCNN(in_channels=3, out_channels=3) self._smoke_test(model, in_channels=3)
def test_TinyCNN(self): model = models.TinyCNN(in_channels=1) self._smoke_test(model)