RuntimeError: Both events must be recorded before calculating elapsed time

Traceback (most recent call last):
File “train.py”, line 531, in
main() # pylint: disable=no-value-for-parameter
File “/home/octa/anaconda3/envs/venvdiffgan/lib/python3.7/site-packages/click/core.py”, line 1130, in call
return self.main(*args, **kwargs)
File “/home/octa/anaconda3/envs/venvdiffgan/lib/python3.7/site-packages/click/core.py”, line 1055, in main
rv = self.invoke(ctx)
File “/home/octa/anaconda3/envs/venvdiffgan/lib/python3.7/site-packages/click/core.py”, line 1404, in invoke
return ctx.invoke(self.callback, **ctx.params)
File “/home/octa/anaconda3/envs/venvdiffgan/lib/python3.7/site-packages/click/core.py”, line 760, in invoke
return __callback(*args, **kwargs)
File “/home/octa/anaconda3/envs/venvdiffgan/lib/python3.7/site-packages/click/decorators.py”, line 26, in new_func
return f(get_current_context(), *args, **kwargs)
File “train.py”, line 524, in main
subprocess_fn(rank=0, args=args, temp_dir=temp_dir)
File “train.py”, line 357, in subprocess_fn
training_loop.training_loop(rank=rank, **args)
File “/home/octa/diffusion-gan/Diffusion-GAN/diffusion-stylegan2/training/training_loop.py”, line 437, in training_loop
value = phase.start_event.elapsed_time(phase.end_event)
File “/home/octa/anaconda3/envs/venvdiffgan/lib/python3.7/site-packages/torch/cuda/streams.py”, line 177, in elapsed_time
return super(Event, self).elapsed_time(end_event)
RuntimeError: Both events must be recorded before calculating elapsed time.

The issue seems to be specific to Diffusion-GAN and solved here.