lightly-ai/lightly

Proper distributed tests

Open

#1.982 geöffnet am 14. Juli 2026

Auf GitHub ansehen
 (3 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (337 Forks)auto 404
advancedenhancementhelp wanted

Repository-Metriken

Stars
 (3.775 Stars)
PR-Merge-Metriken
 (PR-Metriken ausstehend)

Beschreibung

We do some proper distributed tests in tests/utils/test_dist__gather__losses.py, however those are separate from the test modules of the individual losses so adding new losses always gets forgotten or we try to mock the distributed behaviour, which often does not work very well, e.g. in #1980 .

Instead, it would be nice to have a shared distributed context that we could submit distributed tests to. This should run entirely on CPU (gloo backend for PyTorch). A really nice example for this is torchmetrics.

Let me know if you'd like to work on this and I will give some more proper guidance and detail on how this can be done.

Contributor Guide