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)