Task - fix ZeroDivisionError: division by zero for diffusion forecast engine

[sbAsma]

Describe the task. It can be a feature, documentation, etc.

Error message:

5:   File "/p/project1/hclimrep/semcheddine1/WeatherGenerator/src/weathergen/run_train.py", line 190, in train_with_args
5:     trainer.run(cf, devices)
5:   File "/p/project1/hclimrep/semcheddine1/WeatherGenerator/src/weathergen/train/trainer.py", line 452, in run
5:     self.cf.istep
5: ZeroDivisionError: division by zero

happening at: https://github.com/moritzhauschulz/WeatherGenerator/blob/b3b6596721cbc1e9949bbd506347e8f1263b58e3/src/weathergen/train/trainer.py#L431

Already reported. Needed the issue to link the PR.

Hedgedoc URL, if you are keeping notes, plots, logs in hedgedoc.

No response

Area

• datasets, data readers, data preparation and transfer
• model
• science
• infrastructure and engineering
• evaluation, export and visualization
• documentation

[sbAsma]

assign me

[github-actions]

Issue assigned to @sbAsma.

[clessig]

The root cause is elsewhere but it's probably still good to change the line to:

mini_epoch_base = int(self.cf.istep / max(1,len(self.data_loader)))