eloialonso / diamond Goto Github PK

can you provide the code of calculate results/data/DIAMOND.json

Dear Authors,

I noticed that you also omplemented DDPM sampling method in your paper. Could you upload those codes as well.

Sincerely.

Hi,

I met some problems with the diffusion model training. For my project I resize the game frame to gray scale with size 84*84.
This is the reconstruction during early stage training.
And this is the reconstruction result as training goes on, where the details on the image are fading out.

Could you give me some advice on this problem?

Sincerely.

Hi, thanks for such a great job!
You mentioned in your paper that "As a reference, unconstrained Atari agents are usually trained for 50 million steps" and "Each run utilized around 12GB of VRAM and took approximately 2.9 days on a single Nvidia RTX 4090".

I noticed that in your code, the outer loop that calls self.train_agent() is called 1000 times. The inner loop that trains the agent self.train_component(name, steps) is called 5000 times.
This is equivalent to the agent training 1000*5000 steps. The time corresponding to the inner loop is [496/5000 [12:59<1:09:31, 1.08it/s] about 1 hours.
So I guess the total time is about 1000 hours, and this is just for training agent model.
Did I overlook something ? And what should I do to reproduce the training time? @eloialonso @AdamJelley

Dear authors,

Thanks for sharing the code! I did not find the code to train the diffusion model. Could you upload those code? Thanks!

Sincerely