rosinusserrano / pml_vqvae

Repository for the course "Project Machine Learning" during WiSe 24/25 at TU Berlin consisting of a replication of the paper "Neural Discrete Representation Learning" (van den Oord et al., 2018).
0 stars 0 forks source link

Config file and cli parameters #9

Closed timonpalm closed 1 week ago

timonpalm commented 2 weeks ago

Hab jetzt mal eine version mit config file gebaut. Das config file ist die default konfiguration. Es können aber auch alle (vllt hab ich welche vergessen) Parameter über die cli geändert werden. Jetzt haben wir denke ich gute flexibilität aber auch ne easy base zum trainieren.

Hab vorher mit den updates von marco gemerged. Kann sein das da was schief gegangen ist.

Ich finde wir müssen unbedingt die train function aufräumen. Ist, zumindest für mich ziemlich unübersichtlich geworden. Vllt eine train_one_epoch function und eine test function seperat?