diff --git a/train.py b/train.py index e5f06ee4584b22f4389d06ab01a62785f386f154..cc4b9d6ea071c86898cfc29a82324bbad49af576 100644 --- a/train.py +++ b/train.py @@ -83,7 +83,7 @@ model=model.cuda() loss_fn = torch.nn.CrossEntropyLoss() -optimizer=torch.optim.Adam(model.parameters(), lr=0.001) +optimizer=torch.optim.Adam(model.parameters(), lr=0.0001) num_epochs=20 args=types.SimpleNamespace(dry_run=False, log_interval=16)