From 59661f88a3047d0599488a605bffd911e183d0c8 Mon Sep 17 00:00:00 2001 From: davinnovation Date: Mon, 1 Aug 2022 19:03:50 +0900 Subject: [PATCH] Fix typo in EfficientNet-example --- notebooks/EfficientNet-example.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/EfficientNet-example.ipynb b/notebooks/EfficientNet-example.ipynb index 518254c3ab..053f3c8815 100644 --- 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", ")" ]