Fix optimizer loading + Improve configuration for imbalanced categories#17

Open

ivandzen wants to merge 3 commits intodhlab-epfl:masterfrom

qarmir:fixes_and_improvements

Commits on Nov 8, 2025

1. Fix - fix loading optimizer (must be loaded to the same device as model)
ivandzen
committed
Improvement - add weights to ColorLabels - it helps to configure weights for imbalanced categories (in case of rare occurrences or if areas are small comparing to others). This weights are propagat…
ivandzen
committed

Commits on Nov 10, 2025

add weights into inference model
ivandzen
committed