up

mindee · Apr 4, 2022 · 6976419 · 6976419
1 parent f52fa83
commit 6976419
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: