From 393a9840c0337f3fdb33f5b720de91b4490931d1 Mon Sep 17 00:00:00 2001 From: Graham Billey Date: Wed, 8 Jan 2025 16:33:48 -0500 Subject: [PATCH] Fix gradient_clip value. Update gradient_clip=0.01 to gradient_clip=0.1. --- docs/source/tutorials/ar.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/tutorials/ar.ipynb b/docs/source/tutorials/ar.ipynb index 6689dc0e..7cbf0fbc 100644 --- a/docs/source/tutorials/ar.ipynb +++ b/docs/source/tutorials/ar.ipynb @@ -268,7 +268,7 @@ ], "source": [ "pl.seed_everything(42)\n", - "trainer = pl.Trainer(accelerator=\"auto\", gradient_clip_val=0.01)\n", + "trainer = pl.Trainer(accelerator=\"auto\", gradient_clip_val=0.1)\n", "net = NBeats.from_dataset(training, learning_rate=3e-2, weight_decay=1e-2, widths=[32, 512], backcast_loss_ratio=0.1)" ] }, @@ -448,7 +448,7 @@ " max_epochs=3,\n", " accelerator=\"auto\",\n", " enable_model_summary=True,\n", - " gradient_clip_val=0.01,\n", + " gradient_clip_val=0.1,\n", " callbacks=[early_stop_callback],\n", " limit_train_batches=150,\n", ")\n",