Ich hatte die Möglichkeit, GAN zu lernen und wollte trotzdem neue Technologien kennenlernen. Deshalb habe ich mich entschlossen, Style GAN auf Implementierungsebene zu verstehen. Das Versionsproblem von TensorFlow und CUDA hat mich frustriert. Als ich nachforschte, ob Code von PyTorch implementiert wurde, fand ich den folgenden Artikel und beschloss, den Code hier zu laden.
Der Code war leicht zu verstehen,
Also habe ich gespielt, indem ich Docstring-Kommentare hinzugefügt habe, während ich einige Klassen neu angeordnet habe.
Für CUDA müssen "Grafiktreiber aktualisieren" und "CUDA Toolkit installieren" in Siehe diesen Artikel, den ich geschrieben habe ausgeführt werden. es gibt. Wenn Sie keine NVIDIA-Grafikkarte haben, können Sie keine GPU-Berechnungen durchführen. Es kann ausgeführt werden, auch wenn eine GPU-Berechnung nicht möglich ist. Es dauert jedoch einige Sekunden, bis ein Stapel übergeben ist. Daher ist es besser, eine GPU-Umgebung leise einzuführen.
Der Quellcode ist hier ↓ GitHub: StyleGanPytorch
Die Einstellungen können über settings.json geändert werden. Mit den Standardeinstellungen wird die Stapelgröße der 7. Ebene auf 2 reduziert, sodass das Lernen nicht ordnungsgemäß verläuft. Erhöhen Sie diesen Wert, wenn Ihre GPU Platz hat. Wenn es größer als die GPU-Spezifikation ist, kann der Speicher nicht gesichert werden und es tritt ein Fehler auf.
Abgesehen von der Klassenreorganisation und dem Kommentarteil ist der Code im obigen Artikel fast derselbe. Es tut mir leid, aber es war eine sehr gute Studie. Übrigens, in Bezug auf den Betrieb des maschinellen Lernens selbst ist die GPU meines PCs GTX750Ti, daher musste ich die Stapelgröße erheblich reduzieren ... Aus diesem Grund endet das Lernen definitiv nicht mit dem beschriebenen Modell (lacht)
Später werde ich versuchen, die Parameter des Modells als Link zu reduzieren, bis ich einen neuen PC kaufe. Ich hoffe ich kann Illustrationen generieren ...
Wenn Sie das Modell verwenden, das im Originalpapier von Style GAN verwendet wird, können Sie meines Erachtens kein Reproduktionsexperiment durchführen, es sei denn, Sie lassen Geld wie heißes Wasser in die Wolke fallen, also habe ich von Anfang an aufgegeben. (Das realistische Bilderzeugungssystem ist ein Kampf mit den Spezifikationen ...)
Recommended Posts