Zrobić funkcję zapisującą centroidy i model do jednego pliku .pth, tak aby było możliwe wczytanie modelu z pierwszego i z drugiego etapu
Zrobić funkcję zapisującą centroidy i model do jednego pliku .pth, tak aby było możliwe wczytanie modelu z pierwszego i z drugiego etapu