Higher lr; still wont work (gives false in 100% of tests)

2021-09-21 12:57:55 +02:00 · 2021-09-21 12:57:55 +02:00 · 0c197e3e56
commit 0c197e3e56
parent 054b4494d2
1 changed files with 1 additions and 1 deletions
--- 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)