आखिरकार, आपने मॉडल आर्किटेक्चर और loss फंक्शनों को परिभाषित करने में जो मेहनत की थी, अब उसका फल मिलेगा: यह ट्रेनिंग का समय है! आपका कार्य GAN का ट्रेनिंग लूप इम्प्लीमेंट करना और चलाना है। नोट: लंबे रनटाइम से बचने के लिए पहले बैच के बाद एक break स्टेटमेंट रखा गया है.

दो ऑप्टिमाइज़र disc_opt और gen_opt को Adam() ऑप्टिमाइज़र के रूप में इनिशियलाइज़ किया गया है। आपने पहले जो loss निकालने वाले फंक्शन परिभाषित किए थे, gen_loss() और disc_loss(), वे उपलब्ध हैं। एक dataloader भी आपके लिए तैयार है.

ध्यान रखें कि:

disc_loss() के आर्ग्युमेंट्स हैं: gen, disc, real, cur_batch_size, z_dim.
gen_loss() के आर्ग्युमेंट्स हैं: gen, disc, cur_batch_size, z_dim.

समापन

Create Your Free Account