styling

Author: seidnerj

mltu/torch/callbacks.py

@@ -254,15 +254,15 @@ def __init__(
         self, 
         saved_model_path: str,
         input_shape: tuple,
-        export_params: bool=True,
-        opset_version: int=14,
-        do_constant_folding: bool=True,
-        input_names: list=['input'],
-        output_names: list=['output'],
-        dynamic_axes: dict={'input' : {0 : 'batch_size'}, 
-                            'output' : {0 : 'batch_size'}},
-        verbose: bool=False,
-        metadata: dict=None,
+        export_params: bool = True,
+        opset_version: int = 14,
+        do_constant_folding: bool = True,
+        input_names: list = ['input'],
+        output_names: list = ['output'],
+        dynamic_axes: dict = {'input': {0 : 'batch_size'},
+                              'output': {0 : 'batch_size'}},
+        verbose: bool = False,
+        metadata: dict = None,
         ) -> None:
         """ Converts the model from PyTorch to ONNX format after training.