kzl / decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.
MIT License
2.33k stars 440 forks source link

MemoryError: Unable to allocate 6.57 GiB for an array with shape (7056000000,) and data type uint8 #45

Open leeruibin opened 2 years ago

leeruibin commented 2 years ago

I wonder how to solve the memory? When you do your experiment, do you load all data into the memory as each unzipped observation file is more than 2GB? bdb1c32eef4e5f5d3ecc97a8a5ba67b 799ddb26c11a6b7e7b501cbb2d5baea

newne commented 2 years ago

I have the same issue.