shithub: opus

Download patch

ref: 49014454907d515e3c8ca8b06add78ad74c417d1
parent: c5c214df1b214375fce964949598f5b4405c655e
author: Jan Buethe <jbuethe@amazon.de>
date: Fri Sep 29 10:34:11 EDT 2023 

fixed type in error message


--- a/dnn/torch/weight-exchange/wexchange/torch/torch.py
+++ b/dnn/torch/weight-exchange/wexchange/torch/torch.py
@@ -194,7 +194,7 @@
     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 @@
     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
-- 
⑨