bugfix in Phi module
nn.ModuleList has to be used in order to register layer parameters as module parameters (similar to add_module)
This commit is contained in:
@@ -41,6 +41,8 @@ def test_phi():
|
||||
y = phi(torch.rand(input_dim))
|
||||
y = phi(torch.rand(7,7,7,input_dim))
|
||||
|
||||
assert len(phi.parameters() > 0)
|
||||
|
||||
def test_deepsets():
|
||||
m = ds.DeepSetsModule.from_config(ds_config)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user