Naming manipulation tools to facilitate importing model weights from …

google / objax

Apache License 2.0

769 stars 77 forks source link

Closed david-berthelot closed 4 years ago

david-berthelot commented 4 years ago

…other frameworks and to remap weights between models.

VarCollection has a new rename method to rename its variables.
load_var_collection has new option to rename variables from the data read from the file.
Renamer allows to take a dictionary of strings {source:target}, or a tuple (regex, replacement) or a function (str -> str).

Resolves #42

AlexeyKurakin commented 4 years ago

Looks good. Could you add test for load_var_collection with renamer?