55 Commits

Author	SHA1	Message	Date
Cadene	c93ce35d8c	WIP stats (TODO: run tests on stats + cmpute them)	2024-04-04 16:36:03 +00:00
Cadene	1cdfbc8b52	WIP ... WIP WIP train.py works, loss going down WIP eval.py Fix WIP (eval running, TODO: verify results reproduced) Eval works! (testing reproducibility) WIP pretrained model pusht reproduces same results as torchrl pretrained model pusht reproduces same results as torchrl Remove AbstractPolicy, Move all queues in select_action WIP test_datasets passed (TODO: re-enable NormalizeTransform)	2024-04-04 15:31:03 +00:00
Remi Cadene	45b4ecb727	pre-commit run -a	2024-03-02 15:58:21 +00:00
Remi Cadene	1ae6205269	Add Normalize, non_blocking=True in tdmpc, tdmpc run (TODO: diffusion)	2024-03-02 15:53:29 +00:00
Simon Alibert	c1942d45d3	Fixes for PR #4	2024-03-01 14:59:05 +01:00