Only one LossLatentSSLStudentTeacher currently supported

[clessig]

What happened?

With multiple LossLatentSSLStudentTeacher terms the implementation breaks (at least):

WeatherGenerator/src/weathergen/model/model.py

Line 437 in dfcf3e1

for loss, loss_conf in target_losses.items():

and

WeatherGenerator/src/weathergen/train/target_and_aux_ssl_teacher.py

Line 24 in dfcf3e1

self.postprocess_targets = get_target_postprocessing(

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