Johannes Fischer
367f72ec46
Imrove print output
2021-07-30 18:19:15 +02:00
Arec
fb91ee1a62
Merge branch 'main' of https://github.com/sisl/InteractionImitation into main
2021-07-30 05:17:43 -07:00
Arec
5fcd3f49da
adding hyperopt tuning
2021-07-30 05:17:38 -07:00
Johannes Fischer
bd5e854720
Add collision and avg velocity metrics
2021-07-30 13:35:42 +02:00
Arec
e30d0ab1ba
removing outliers from expert tracks
2021-07-29 08:20:40 -07:00
Arec
3b4ef6ffb5
adding tool for visualizing acceleration distributions, and making nframes an arg
2021-07-29 07:35:21 -07:00
Johannes Fischer
afc3719ab9
Fix bug with wrong argument order
2021-07-29 15:30:55 +02:00
Johannes Fischer
3b623a1467
Add testing script for raytune experiments
2021-07-29 15:30:41 +02:00
Johannes Fischer
35634fd2eb
Separate experiment from main.py
2021-07-29 14:40:19 +02:00
Arec
e47d69dbc1
changing relative state dim to 6
2021-07-29 02:54:41 -07:00
Arec
fdbdc7f9f0
making data processing happen on front end, not on data loader. saving a ton of time
2021-07-29 02:53:20 -07:00
Arec
09f77e0587
changing how tune reporting works so the scheduler doesnt break if itcant find cv loss. also fixed bug in config structure that was rendering impossible policies
2021-07-28 09:55:11 -07:00
Arec
5a7090a21c
Merge branch 'main' of https://github.com/sisl/InteractionImitation into main
2021-07-27 14:32:52 -07:00
Arec
87aa19b86b
getting hyperparameter tunning with ray tune working. updating default network with optimization and general parameters.
2021-07-27 14:30:28 -07:00
Johannes Fischer
2bd638c247
Test for batch data in test_deepsets_policy
2021-07-27 12:37:40 +02:00
Arec
ca80fa19eb
Merge branch 'main' of https://github.com/sisl/InteractionImitation into main
2021-07-26 07:52:19 -07:00
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
Johannes Fischer
5e89a83c88
Fix tests
2021-07-26 16:35:45 +02: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
Arec
7b2ca6edc7
adding tensorboard writer for training loss and cv loss
2021-07-26 02:38:50 -07:00
Arec
969812c5cc
Merge branch 'main' of https://github.com/sisl/InteractionImitation into main
2021-07-23 04:44:57 -07:00
Arec
5b09374c77
getting training and testing loop working, adding tqdm to simulator, and reduced number of frames, updating readme
2021-07-23 04:44:50 -07: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
5c15ec2db1
Merge branch 'main' of https://github.com/sisl/InteractionImitation into main
2021-07-23 01:28:07 -07:00
Arec
9c7fc47abe
updating test batch
2021-07-23 01:27:57 -07:00
Johannes Fischer
5d3b3fa1bc
Merge branch 'main' of github.com:sisl/InteractionImitation
2021-07-23 09:30:26 +02:00
Johannes Fischer
9b0aeb182f
Add .vscode to gitignore
2021-07-23 09:30:23 +02:00
Arec
563a2cfbd4
making a differentiable transform for use for pytorch, making sure the fitting function treats nans properly while fitting. next issue: forward pass is returning nans
2021-07-22 12:37:36 -07:00
Arec
08eb898812
adding dtypes and fixing matrix indexing
2021-07-22 12:06:42 -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
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
c939a8e131
Remove todo
2021-07-20 18:45:49 +02: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
8534e8a437
Update scratch space
2021-07-20 15:25:07 +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
