train error

[XLR-man]

i download pretrained diffusion model you provided，but when i train，it got error：

  File "/data/xielangren/project/Aleth-NeRF/src/model/llnerf/model.py", line 81, in setup
    patch_diffusion_model = load_patch_diffusion_model(Path(self.patch_regulariser_path))
  File "/data/xielangren/project/Aleth-NeRF/src/model/llnerf/learned_regularisation/patch_regulariser.py", line 138, in load_patch_diffusion_model
    trainer.load(str(reg_checkpoint_path))
  File "/data/xielangren/project/Aleth-NeRF/src/model/llnerf/learned_regularisation/diffusion/denoising_diffusion_pytorch.py", line 836, in load
    self.ema.load_state_dict(data['ema'])
AttributeError: 'Trainer' object has no attribute 'ema'

how to fix the error

[XLR-man]

I found that this seems to be the case for multi-GPU training, but not for single-GPU training. Why is that?