Commit graph

9 commits

Author SHA1 Message Date
Ruben van de Ven
51d6157af9 Refactor Trajectron++ to work as module and expose training command 2023-12-06 12:28:56 +01:00
Ruben van de Ven
4fa3ce95ee Trajectron as a module while supporting old models 2023-10-09 21:06:36 +02:00
Ruben van de Ven
dfa1d43f2e Change imports to support usage as module
Note that this does require rerunning the `process_data.py` scripts in the example folders so that the dill-files are updated.
2023-10-09 20:27:29 +02:00
Ruben van de Ven
f4d860907c Fix compatibility with numpy > 1.20 2023-10-04 21:16:02 +02:00
Ruben van de Ven
d182787c83 Fix compatibility with python 3.10 2023-10-04 21:14:46 +02:00
Boris Ivanovic
58b9763d02
Small batch preprocessing fix
Ensuring that the robot future standardization is with respect to a node's current position.
2022-06-14 17:44:35 -04:00
BorisIvanovic
03712b70a4 Fixed the gradient calculations, going to make another big update to the codebase soon but wanted to get this out first. Fixes #40 and #26. 2021-08-28 18:11:08 -07:00
BorisIvanovic
174198bd4f Adding ability to run on streaming data. 2020-12-09 22:42:06 -05:00
BorisIvanovic
40dec06e9c Updated codebase following the paper update. 2020-04-05 21:43:49 -04:00