1. Nauka
    2. /
  2. Kursy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

ćwiczenie

Diskriminátor

Když máš generátor hotový, dalším krokem při stavbě GANu je sestavení diskriminátoru. Ten přijímá výstup generátoru jako vstup a produkuje binární predikci: je vstup vygenerovaný, nebo skutečný?

torch.nn je už naimportovaný jako nn. Máš také k dispozici pomocnou funkci disc_block(), která vrací blok složený z lineární vrstvy a aktivační funkce LeakyReLU. Použij ji jako stavební kámen diskriminátoru.

def disc_block(in_dim, out_dim):
    return nn.Sequential(
        nn.Linear(in_dim, out_dim),
        nn.LeakyReLU(0.2)
    )

Instrukcje

100 XP
  • Přidej poslední blok diskriminátoru do modelu se správnou velikostí vstupu a výstupem 256.
  • Za posledním blokem diskriminátoru přidej lineární vrstvu, která namapuje výstup na velikost 1.
  • Definuj metodu forward() tak, aby vstupní obrázek prošel sekvenčním blokem definovaným v __init__().