From 0c197e3e560f8d0844bb2e0e9efb6bacf57b0e46 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 21 Sep 2021 12:57:55 +0200 Subject: [PATCH] Higher lr; still wont work (gives false in 100% of tests) --- discriminator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/discriminator.py b/discriminator.py index 99c1002..a185f96 100644 --- a/discriminator.py +++ b/discriminator.py @@ -36,7 +36,7 @@ def train(model, seq_len=16*64): model.train() criterion = nn.BCELoss() - optimizer = optim.Adam(model.parameters(), lr=0.01) + optimizer = optim.Adam(model.parameters(), lr=0.1) for epoch in range(1024): state_h, state_c = model.init_state(seq_len)