diff --git a/dnn/torch/weight-exchange/wexchange/torch/torch.py b/dnn/torch/weight-exchange/wexchange/torch/torch.py index 35723c22909cf381f26a84e4fccfb39aba20402a..6befe9f46fbe0451bc9e2a631b8a744b42a1c06d 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