Update trainer.py

trainer.py

@@ -270,7 +270,7 @@ if __name__ == "__main__":
         tr_loss = train_one_epoch(
             model, train_loader, optimizer, device=device,
             scheduler=scheduler, grad_accum_steps=grad_accum,
-            amp=True, max_grad_norm=1.0,
+            amp=amp, max_grad_norm=max_grad_norm,
         )
         dev_loss, dev_f1 = eval_loss_and_token_f1(model, dev_loader, id2label, device=device)