niftynet.engine.handler_model module¶
This module implements a model checkpoint loader and writer.
-
make_model_name
(model_dir)[source]¶ Make the model checkpoint folder. the checkpoint file will be located at model_dir/models/ folder, the filename will start with FILE_PREFIX.
Parameters: model_dir – niftynet model folder Returns: a partial name of a checkpoint file model_dir/model/FILE_PREFIX
-
class
ModelRestorer
(model_dir, initial_iter=0, is_training_action=True, vars_to_restore=None, **_unused)[source]¶ Bases:
object
This class handles restoring the model at the beginning of a session.
-
class
ModelSaver
(model_dir, save_every_n=0, max_checkpoints=1, is_training_action=True, **_unused)[source]¶ Bases:
object
This class handles iteration events to save the model as checkpoint files.
-
init_saver
(_sender, **_unused)[source]¶ Initialise a model saver.
Parameters: - _sender –
- _unused –
Returns:
-