HLCaptain / ssl-in-the-wild

Comparison of Self-Supervised Learning models.

0 stars 2 forks source link

VICReg train Docker integration #21

Closed zeef69 closed 10 months ago

zeef69 commented 10 months ago

11 Elsőfelének megoldása

VICReg SSL tanítás integrálva Dockerbe.

birds_datamodule transzformációval kiegészítve
vicreg_module model hozzáadva
config fájlok elkészítve az adatmodulhoz és modelhez
- callback implementálva
train configfájl default értékként ezeken tanít

Trainből ki lett szedve a default értékként a tesztelés, külön kell test=True-ra beállítani.

Tesztelve cpu-n, kis epoch méreten. Csak a fájl generálás volt a cél.

Classifier tanítás és tesztelés integrálása Dockerbe.

classifier_datamodule elkészítes a birds_datamodule mintájára (eltérés: transzformációk és DataLoader-ek)
classifier_module model hozzáadva TODO model.backbone beolvasás nem működik
config fájlok elkészítve az adatmodulhoz és modelhez
- callback implementálva
eval configfájl default értékként ezeken tesztel

Továbbbi javítás és tesztelést igényel:

Classifier tanítás és tesztelés.
használati útmutató írása

zeef69 commented 10 months ago

22 Classifier edit, num_workers=1 Commithoz kapcsolódó bug