Johannes Fischer
5e89a83c88
Fix tests
2021-07-26 16:35:45 +02:00
Johannes Fischer
8a4094037a
update forward pass test
2021-07-23 11:16:08 +02:00
Arec
acd2b730a6
adding transforms to test case
2021-07-23 01:38:20 -07:00
Arec
9c7fc47abe
updating test batch
2021-07-23 01:27:57 -07:00
Johannes Fischer
1c22bd6111
bugfix in deepsets
2021-07-22 18:44:12 +02:00
Arec
91d052445e
making test case for typing bug and fixing some small typing errors in bc
2021-07-22 08:23:46 -07:00
Johannes Fischer
827a8e7172
bugfix in Phi module
...
nn.ModuleList has to be used in order to register layer parameters as module parameters (similar to add_module)
2021-07-22 14:33:05 +02:00
Johannes Fischer
c939a8e131
Remove todo
2021-07-20 18:45:49 +02:00
Johannes Fischer
61f06c95c2
Improve deepsets module
...
module can now deal with nan values for nonexisting relative states
in case all relative states are nan, the latent representation is zeroed, which is consitent with an empty sum
2021-07-20 18:44:07 +02:00
Johannes Fischer
226a427436
Adapt policy input format
2021-07-20 15:45:15 +02:00
Johannes Fischer
ee7b6b607f
Fix deprecation warning in sigmoid
2021-07-20 12:52:02 +02:00
Johannes Fischer
49e32fe37b
Add test and example config for DeepSetsPolicy
2021-07-19 18:38:50 +02:00
Johannes Fischer
0e6b1e102e
Update deepsets phi module to allow different final activation
2021-07-19 18:36:52 +02:00
Johannes Fischer
ac5bbce72a
Implement deepsets and deepsets policy, add tests
2021-07-19 17:28:42 +02:00
Arec
b684a5db6d
making initial file shell
2021-07-16 07:15:33 -07:00
