kzl / decision-transformer

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

torch.stack bug #62

Open ChaselWan opened 1 year ago

ChaselWan commented 1 year ago

Hi! Thanks for your great work! when I run training, the error shows "we got [30, xxx] ( xxx :I forgot the number ), but got [30, 0] at entry 8", but when I set num_workers = 0 , it works training. How can I fix this problem?