allow history observation

2024-09-11 21:44:47 -04:00
parent 2ddf63b8f5
commit f13eb203e1
26 changed files with 64 additions and 23 deletions
@@ -1,8 +1,6 @@
 """
 Pre-training data loader. Modified from https://github.com/jannerm/diffuser/blob/main/diffuser/datasets/sequence.py

-TODO: implement history observation
-
 No normalization is applied here --- we always normalize the data when pre-processing it with a different script, and the normalization info is also used in RL fine-tuning.

 """