I got issue during training

[Simon006]

creating model and diffusion...
creating data loader...
training...
Traceback (most recent call last):
File "D:\conditional-polyp-diffusion-main\improved-diffusion\scripts\image_train.py", line 83, in
main()
File "D:\conditional-polyp-diffusion-main\improved-diffusion\scripts\image_train.py", line 41, in main
TrainLoop(
File "D:\conditional-polyp-diffusion-main\improved-diffusion\scripts\improved_diffusion\train_util.py", line 78, in init
self._load_and_sync_parameters()
File "D:\conditional-polyp-diffusion-main\improved-diffusion\scripts\improved_diffusion\train_util.py", line 127, in _load_and_sync_parameters
dist_util.sync_params(self.model.parameters())
File "D:\conditional-polyp-diffusion-main\improved-diffusion\scripts\improved_diffusion\dist_util.py", line 75, in sync_params
dist.broadcast(p, 0)
File "C:\Users\OMEN\miniconda3\lib\site-packages\torch\distributed\distributed_c10d.py", line 1408, in broadcast
work.wait()
RuntimeError: a leaf Variable that requires grad is being used in an in-place operation.