search

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

Dataset by class #12

Closed timonpalm closed 4 days ago

timonpalm commented 6 days ago

Sollte funktionieren, hab aber ehrlich gesagt nicht mehr ausführlich getestet.

Denke das sollte reichen, dann können wir mit cifar einmal seperat auf jeder klasse testen und gucken, ob es da Unterschiede gibt

Edit: Hab auch die zu kleinen Bilder jetzt rausgefiltert. Sind nur 41.

KonstantinAusborn commented 5 days ago

Was ist noch mit dem conflict? Hat sich irgendwas mit der Batch size verändert? Die wollen wir ja eigentlich behalten :D

rosinusserrano commented 5 days ago

Was ist noch mit dem conflict? Hat sich irgendwas mit der Batch size verändert? Die wollen wir ja eigentlich behalten :D

Die batch_size wurde nicht von den cli args gelesen und ich habe das dann noch schnell auf die mein gepusht. Es können einfach beide lines (also batchsize und classidx akzeptiert werden)

timonpalm commented 5 days ago

Ja dann, good to go! Kann den iwie nicht resolven