diff --git a/unit2/01_finetuning_and_guidance.ipynb b/unit2/01_finetuning_and_guidance.ipynb index 715867b..9074801 100644 --- a/unit2/01_finetuning_and_guidance.ipynb +++ b/unit2/01_finetuning_and_guidance.ipynb @@ -597,7 +597,7 @@ ], "source": [ "num_epochs = 2 # @param\n", - "lr = 1e-5 # 2param\n", + "lr = 1e-5 # @param\n", "grad_accumulation_steps = 2 # @param\n", "\n", "optimizer = torch.optim.AdamW(image_pipe.unet.parameters(), lr=lr)\n",