Merge pull request #1217 from davinnovation/patch-1

peri044 · web-flow · commit db61e904f5e9 · 2022-08-01T09:35:32.000-04:00
Fix typo in EfficientNet-example
diff --git a/notebooks/EfficientNet-example.ipynb b/notebooks/EfficientNet-example.ipynb
@@ -611,7 +611,7 @@
     "# The compiled module will have precision as specified by \"op_precision\".\n",
     "# Here, it will have FP16 precision.\n",
     "trt_model_fp16 = torch_tensorrt.compile(model, inputs = [torch_tensorrt.Input((128, 3, 224, 224), dtype=torch.half)],\n",
-    "    enabled_precisions = {torch.half}, # Run with FP32\n",
+    "    enabled_precisions = {torch.half}, # Run with FP16\n",
     "    workspace_size = 1 << 22\n",
     ")"
    ]

Original file line number	Diff line number	Diff line change
`@@ -611,7 +611,7 @@`
`611`	`611`	`"# The compiled module will have precision as specified by \"op_precision\".\n",`
`612`	`612`	`"# Here, it will have FP16 precision.\n",`
`613`	`613`	`"trt_model_fp16 = torch_tensorrt.compile(model, inputs = [torch_tensorrt.Input((128, 3, 224, 224), dtype=torch.half)],\n",`
`614`		`- " enabled_precisions = {torch.half}, # Run with FP32\n",`
	`614`	`+ " enabled_precisions = {torch.half}, # Run with FP16\n",`
`615`	`615`	`" workspace_size = 1 << 22\n",`
`616`	`616`	`")"`
`617`	`617`	`]`