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:
Johannes Fischer
2021-07-22 14:33:05 +02:00
parent 1ca9914bf9
commit 827a8e7172
2 changed files with 3 additions and 4 deletions

View File

@@ -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)