LightZero icon indicating copy to clipboard operation
LightZero copied to clipboard

Replicating multi-GPU EfficientZero Atari results

Open selfsim opened this issue 1 month ago • 1 comments

Hello,

I want to replicate all E0 Atari results to compare my custom learner against.

I have SSH access to 32 CPU core + 4 GPU nodes on a cluster. I managed to get the single GPU atari_efficientzero_config.py to work, but I have been unable to run the multi-GPU atari_efficientzero_multigpu_ddp_config.py successfully. Can you confirm that the latter works on the current version? And if so, can you provide the environment details (python version etc.) and the execution instructions used?

Also, do you have the hyperparameters for the E0 Atari trials? I'm assuming they are all default but it would be nice to have confirmation.

selfsim avatar May 10 '24 23:05 selfsim