eloialonso/diamond
DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.
PythonMIT
Issues
- 0
The traning data can't be downloaded.
#22 opened by PeiPeiWow - 4
The training data set confuses me, how to build it?
#19 opened by win10ogod - 0
Bug of restarting training?
#24 opened by kxhit - 8
Packages issue
#8 opened by f1amy - 0
- 2
Can't train with "python src/main.py env.train.id=BreakoutNoFrameskip-v4 common.device=cuda:0"
#20 opened by 1846333564 - 0
Training the diffusion model on my own data
#21 opened by ritvikpasham - 18
How can I enable GPU?
#9 opened by Crimsonfart - 1
this is AWESOME 🎉🎉🎉
#18 opened by johnpaulbin - 3
Training set size?
#12 opened by danzeeeman - 1
would it be possible to use pyinstaller to make it more userfriendly?
#17 opened by thespecificjhevin - 2
great research, how can i train on my own dataset? what do i need to prepare? thank you very much
#15 opened by D-Mad - 2
Question about training
#7 opened by alhojel - 1
Not utilizing 4090 GPU?
#14 opened by GitInMotion - 1
- 1
AssertionError
#10 opened by andreykuzovlevv - 1
- 1
DDPM sampling code
#5 opened by SliencerX - 2
- 2
- 1
can you provide the code?
#2 opened by ZhikanggFu - 1
About training time
#1 opened by hutchinsonian