Fix int fill-value error in test_modules.py

This commit is contained in:
Jensun Ravichandran 2020-07-30 11:42:37 +02:00
parent 9ccbec52f7
commit b138277608

View File

@ -199,7 +199,7 @@ class TestPrototypes(unittest.TestCase):
def test_prototypes1d_func_initializer(self): def test_prototypes1d_func_initializer(self):
def my_initializer(*args, **kwargs): def my_initializer(*args, **kwargs):
return torch.full((2, 99), 99), torch.tensor([0, 1]) return torch.full((2, 99), 99.0), torch.tensor([0, 1])
p1 = prototypes.Prototypes1D(input_dim=99, p1 = prototypes.Prototypes1D(input_dim=99,
nclasses=2, nclasses=2,