location_weight should be initialized in constructor and not moved to device for every batch

[clessig]

What happened?

See

WeatherGenerator/src/weathergen/train/loss_calculator.py

Line 123 in a6e07a4

weights_locations = weights_locations.to(device=self.device, non_blocking=True)

location_weight should be initialized in constructor and not moved to device for every batch

What are the steps to reproduce the bug?

No response

Hedgedoc link to logs and more information. This ticket is public, do not attach files directly.

No response