Commit Graph

30 Commits

Author SHA1 Message Date
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
1ca9914bf9 fixing bugs in transform, expert demo processing, main train function, and behavior cloning class. need to get bc class parameters to return nonempty list 2021-07-21 09:44:20 -07:00
Arec
5758af5dd8 Merge branch 'main' of https://github.com/sisl/InteractionImitation into main 2021-07-21 08:22:15 -07:00
Arec
18b8e0c58f updating main testing function to use configs and seeds, finishing first pass at behavior cloning policy and training loop. not yet tested 2021-07-21 08:22:08 -07:00
Arec
4350cf8cd5 defining transform class to do tensor size manipulation before and after transform 2021-07-21 08:20:55 -07:00
Arec
3422e9c9ef exporting policy class and letting default final activation do nothing 2021-07-21 08:20:15 -07:00
Johannes Fischer
ca0f520c89 Merge branch 'main' of github.com:sisl/InteractionImitation 2021-07-20 18:44:11 +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
ffdff12ccb Update deepsets to deal with nans (first version) 2021-07-20 18:23:51 +02:00
Arec
6794b4cad8 making saving and loading functions class requirements, working on behavior cloning policy class and training function 2021-07-20 08:29:00 -07:00
Arec
1a74fa5237 making general-purpose metric function 2021-07-20 07:05:09 -07:00
Johannes Fischer
226a427436 Adapt policy input format 2021-07-20 15:45:15 +02:00
Johannes Fischer
7109afb21f Merge branch 'main' of github.com:sisl/InteractionImitation 2021-07-20 15:25:10 +02:00
Johannes Fischer
ba10a1f53b Implement transforms for normalization 2021-07-20 15:24:34 +02:00
Arec
a15e8c29ff moving transforms out of dataset class, will be exclusively in policy classes 2021-07-20 06:16:54 -07:00
Arec
2fb5d5e5b1 developing main experiment loop, functions required to implement in bc and other imitation methods 2021-07-20 05:58:24 -07:00
Arec
c0755f5bb5 Merge branch 'main' of https://github.com/sisl/InteractionImitation into main 2021-07-20 05:46:47 -07:00
Arec
2e29f42d75 add flag to process all tracks of a particular location 2021-07-20 05:46:43 -07:00
Johannes Fischer
ee7b6b607f Fix deprecation warning in sigmoid 2021-07-20 12:52:02 +02:00
Arec
793bb64af4 adding data loader class 2021-07-19 13:07:09 -07:00
Johannes Fischer
ea460d07d2 Merge branch 'main' of github.com:sisl/InteractionImitation 2021-07-19 18:39:01 +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
Arec
6964cb0c6c got processing of expert demonstrations functional 2021-07-19 08:53:01 -07:00
Johannes Fischer
3cd43d8c3a Merge branch 'main' of github.com:sisl/InteractionImitation 2021-07-19 17:28:49 +02:00
Johannes Fischer
ac5bbce72a Implement deepsets and deepsets policy, add tests 2021-07-19 17:28:42 +02:00
Arec
8acfbc7993 updating expert data to keep track of deviations and check for collisions 2021-07-19 07:10:28 -07:00
Arec
2373d924b7 adding expert data loader and saver. same issue as previous commit, unable to get through full simulation with backtracked accelerations 2021-07-16 08:33:40 -07:00
Arec
a6e691bd1a working on expert data saver and loader. mostly working, but issue getting all the way through simulation with backtracked accelerations 2021-07-16 08:32:48 -07:00
Arec
b684a5db6d making initial file shell 2021-07-16 07:15:33 -07:00