update

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)