Currently, all the models are hardcoded in models.py. This is an issue when loading old models because TF checkpoints requires the structure of the model to match that as saved in the checkpoint.
Add the ability to load a model architecture from a config.
Currently, all the models are hardcoded in
models.py
. This is an issue when loading old models because TF checkpoints requires the structure of the model to match that as saved in the checkpoint.Add the ability to load a model architecture from a config.