dodox/shark
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source
This commit is contained in:
Dominik Moritz Roth 2021-09-21 12:57:55 +02:00
parent 054b4494d2
commit 0c197e3e56
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
discriminator.py
View File

@ -36,7 +36,7 @@ def train(model, seq_len=16*64):
model.train() model.train()
criterion = nn.BCELoss() criterion = nn.BCELoss()
optimizer = optim.Adam(model.parameters(), lr=0.01) optimizer = optim.Adam(model.parameters(), lr=0.1)
for epoch in range(1024): for epoch in range(1024):
state_h, state_c = model.init_state(seq_len) state_h, state_c = model.init_state(seq_len)