Fix int fill-value error in test_modules.py
This commit is contained in:
parent
9ccbec52f7
commit
b138277608
@ -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,
|
||||||
|
Loading…
Reference in New Issue
Block a user