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).
War doch iwie schwieriger als gedacht, das halbwegs clean einzubauen. Sollte jetzt aber funktionieren.
Ihr braucht dafür einen wandb account und müsst export WANDB_API_KEY=<euer key> machen. Wenn ihr den Projektnamen nicht ändert und ich euch ins Projekt eingeladen haben sollten wir alle runs sehen.
Ein paar Überlegungen:
da wir jetzt alles über wandb loggen, weiß ich nicht, ob wir überhapt noch die Funktionalität zum lokalen plotten und visualisieren brauchen. Die Vis Funktion für den Datensatz war eine Anforderung und muss bleiben.
Iwie hatte ich viel mit den unterschiedlichen Losses zu Kämpfen und finde die Stats_keeper Klasse auch ziemlich unübersichtlich. Vllt kann man das dann mit dem wandb_wrapper kombinieren?
Ich finde es iwie unsauber, dass die train und test Funktion den letzten output und Batch zurückgeben, die dann geplotten werden. Funktioniert soweit aber KA
War doch iwie schwieriger als gedacht, das halbwegs clean einzubauen. Sollte jetzt aber funktionieren.
Ihr braucht dafür einen wandb account und müsst
export WANDB_API_KEY=<euer key>machen. Wenn ihr den Projektnamen nicht ändert und ich euch ins Projekt eingeladen haben sollten wir alle runs sehen.Ein paar Überlegungen: