Arec
d9436daaba
changing default network, making deepsets network choose output dimension appropriately, making bc config to be called by ray
2021-07-26 07:52:11 -07:00
Arec
69359b5af3
periodically savingin out model and adding functionality to make identity Phi networks (for 0-dim NNs)
2021-07-26 05:51:08 -07:00
Johannes Fischer
1c22bd6111
bugfix in deepsets
2021-07-22 18:44:12 +02: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
Arec
5758af5dd8
Merge branch 'main' of https://github.com/sisl/InteractionImitation into main
2021-07-21 08:22:15 -07:00
Arec
3422e9c9ef
exporting policy class and letting default final activation do nothing
2021-07-21 08:20:15 -07: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
ffdff12ccb
Update deepsets to deal with nans (first version)
2021-07-20 18:23:51 +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
