From 6976419b71e4fe18f0d390f5739c6fd2dce601f6 Mon Sep 17 00:00:00 2001 From: felix Date: Mon, 4 Apr 2022 19:43:38 +0200 Subject: [PATCH] up --- doctr/models/utils/pytorch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doctr/models/utils/pytorch.py b/doctr/models/utils/pytorch.py index 757a4f5604..bf1f798c0c 100644 --- a/doctr/models/utils/pytorch.py +++ b/doctr/models/utils/pytorch.py @@ -92,8 +92,8 @@ def export_classification_model_to_onnx(model: nn.Module, exp_name: str, dummy_i >>> export_classification_model_to_onnx(model, "my_model", dummy_input=torch.randn(1, 3, 32, 32)) Args: - model: the pytorch model to be exported - exp_name: the name of the exported model + model: the PyTorch model to be exported + exp_name: the name for the exported model dummy_input: the dummy input to the model Returns: