From 7f0a4428efe9fda62a61fd747c36dd7bd670a380 Mon Sep 17 00:00:00 2001 From: Alexander Markovich Date: Tue, 19 Mar 2024 23:18:26 +0100 Subject: [PATCH] Fix week08 homework --- week08_inference_software/homework.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/week08_inference_software/homework.ipynb b/week08_inference_software/homework.ipynb index aa3b665..1312713 100644 --- a/week08_inference_software/homework.ipynb +++ b/week08_inference_software/homework.ipynb @@ -358,7 +358,7 @@ "metadata": {}, "outputs": [], "source": [ - "input_shape = 1, 32, 64, 64\n", + "input_shape = (1, 32, 64, 64)\n", "num_groups = 8\n", "test_group_norm(input_shape, num_groups, torch.float16)" ] @@ -380,8 +380,8 @@ "source": [ "# You need to check your implementation for the following parameters\n", "batch_size = 2\n", - "for image_resolution in [32, 128, 512, 1024, 2048, 4096]:\n", - " for num_channels in [32, 128, 512, 1024]:\n", + "for image_resolution in [32, 128, 512, 1024, 1536, 2048]:\n", + " for num_channels in [32, 128, 386, 512]:\n", " for num_groups in [1, 4, 8, 16, 32]:\n", "\n", " dummy_input = torch.randn(batch_size, num_channels, image_resolution, image_resolution)\n",