arpitbansal297/Cold-Diffusion-Models

AttributeError: 'GaussianDiffusion' object has no attribute 'module'

Amarkr1 opened this issue · 1 comments

Amarkr1 commented

Cold-Diffusion-Models/deblurring-diffusion-pytorch/deblurring_diffusion_pytorch/deblurring_diffusion_pytorch.py

Line 1241 in f8b1379

X_0s, X_ts = self.ema_model.module.all_sample(batch_size=batches, img=og_img, times=s_times) # change for DP

Modified this to:
X_0s, X_ts = self.ema_model.all_sample(batch_size=batches, img=og_img, times=s_times) # change for DP

seems to solve the error.

arpitbansal297 commented

Yes, .module is there because of data parallel.