search

MishaLaskin / curl

CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning
MIT License
561 stars 88 forks source link

retain_graph=True #5

Closed rainbow979 closed 4 years ago

rainbow979 commented 4 years ago

Thanks for you great work. I have a problem when I want to modify the code. It hinted that I must to use retain_graph=True. Where am I wrong perhaps?