Skip to content
Snippets Groups Projects
Unverified Commit 49014454 authored by Jan Buethe's avatar Jan Buethe
Browse files

fixed type in error message

parent c5c214df
No related branches found
No related tags found
No related merge requests found
Pipeline #4098 passed
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment