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