12 Commits

Author	SHA1	Message	Date
Cadene	8e5b4365ac	fix	2024-04-11 13:57:22 +00:00
Cadene	c08003278e	test_examples are passing	2024-04-10 13:45:45 +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
Cadene	4a8c5e238e	issue with cat_and_write_video	2024-03-26 10:12:16 +00:00
Alexander Soare	1a1308d62f	fix environment seeding ... add fixes for reproducibility only try to start env if it is closed revision fix normalization and data type Improve README Improve README Tests are passing, Eval pretrained model works, Add gif Update gif Update gif Update gif Update gif Update README Update README update minor Update README.md Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> Update README.md Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> Address suggestions Update thumbnail + stats Update thumbnail + stats Update README.md Co-authored-by: Alexander Soare <alexander.soare159@gmail.com> Add more comments Add test_examples.py	2024-03-26 10:10:43 +00:00
Alexander Soare	4a48b77540	fix normalization and data type	2024-03-25 09:44:03 +00:00
Remi Cadene	f95ecd66fc	Improve visualize_dataset, Improve AbstractReplayBuffer, Small improvements	2024-03-06 10:15:57 +00:00
Cadene	ae050d2e94	Solve conflicts + pre-commit run -a	2024-02-29 23:31:32 +00:00
Simon Alibert	7e024fdce6	Ran pre-commit run --all-files	2024-02-29 13:37:48 +01:00
Cadene	e3643d6146	Wandb works, One output dir	2024-02-22 12:14:12 +00:00
Cadene	ece89730e6	Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)	2024-02-21 00:49:40 +00:00
Cadene	228c045674	Eval reproduced! Train running (but not reproduced)	2024-02-10 15:46:24 +00:00