From 49014454907d515e3c8ca8b06add78ad74c417d1 Mon Sep 17 00:00:00 2001
From: Jan Buethe <jbuethe@amazon.de>
Date: Fri, 29 Sep 2023 14:34:11 +0200
Subject: [PATCH] fixed type in error message
---
dnn/torch/weight-exchange/wexchange/torch/torch.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/dnn/torch/weight-exchange/wexchange/torch/torch.py b/dnn/torch/weight-exchange/wexchange/torch/torch.py
index 35723c229..6befe9f46 100644
--- a/dnn/torch/weight-exchange/wexchange/torch/torch.py
+++ b/dnn/torch/weight-exchange/wexchange/torch/torch.py
@@ -194,7 +194,7 @@ def dump_torch_weights(where, module, name=None, verbose=False, **kwargs):
elif isinstance(module, torch.nn.Embedding):
return dump_torch_embedding_weights(where, module)
else:
- raise ValueError(f'dump_tf_weights: layer of type {type(module)} not supported')
+ raise ValueError(f'dump_torch_weights: layer of type {type(module)} not supported')
def load_torch_weights(where, module):
""" generic function for loading weights of some torch.nn.Module """
@@ -209,4 +209,4 @@ def load_torch_weights(where, module):
elif isinstance(module, torch.nn.Embedding):
load_torch_embedding_weights(where, module)
else:
- raise ValueError(f'dump_tf_weights: layer of type {type(module)} not supported')
\ No newline at end of file
+ raise ValueError(f'dump_torch_weights: layer of type {type(module)} not supported')
\ No newline at end of file
--
GitLab