To support saving and loading Task, the following changes have been made:
support saving and loading task.json and task.weights.h5.
support saving and loading Preprocessor (added preprocessor.json).
move preset saving and loading logic to base classes, e.g. Tokenizer, Backbone, etc. and only kept the low-level preset manipulation in preset_utils.py.
TODO:
add unit tests for new functions added to preset_utils.py.
add unit test for saving and loading in each base class.
Future plan: currently backbone and config are called config.json and model.weights.h5. Our plan is to rename these to backbone.json and backbone.weights.h5 in a followup PR.
To support saving and loading
Task
, the following changes have been made:task.json
andtask.weights.h5
.Preprocessor
(addedpreprocessor.json
).preset_utils.py
.TODO:
preset_utils.py
.Future plan: currently backbone and config are called
config.json
andmodel.weights.h5
. Our plan is to rename these tobackbone.json
andbackbone.weights.h5
in a followup PR.