Fix dimensions in week08 homework (#23)

Author: markovka17

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",