src.model.deeplearn.handle.dl_path_manager

Classes

DLPathManager(**kwargs)

Class to handle the paths for deep learning models.
class src.model.deeplearn.handle.dl_path_manager.DLPathManager(**kwargs)

Class to handle the paths for deep learning models.

Variables:

  • summary_report_path (str) – Path to the file where the summary report will be written, i.e., the report that summarizes the model’s architecture.

  • training_history_dir (path) – Path to the directory where the training history plots and reports will be exported, i.e., information related to the training along many epochs.

  • checkpoint_path (str) – The path where the model’s checkpoint will be exported. It is used to keep the best model when using the checkpoint callback strategy during training.

  • feat_struct_repr_dir (str) – The path where the information relative to FeaturesStructuringLayer layers will be exported.

  • rbf_feat_extract_repr_dir (str) – The path where the information relative to RBFFeatExtractLayer layers will be exported.

  • rbf_feat_processing_repr_dir (str) – The path where the information relative to RBFFeatProcessingLayer layers will be exported.

  • kpconv_representation_dir (str) – The path where the information relative to KPConvLayer layers will be exported.

  • skpconv_representation_dir (str) – The path where the information relative to StridedKPConvLayer layers will be exported.

  • lkpconv_representation_dir (str) – The path where the information relative to LightKPConvLayer layers will be exported.

  • slkpconv_representation_dir (str) – The path where the information relative to StridedLightKPConvLayer layers will be exported.

  • kpconvx_representation_dir (str) – The path where the information relative to KPConvXLayer layers will be exported.

__init__(**kwargs)

Initialize a DLPathManager.

update_paths(model_args, arch)

Consider the current specification of model handling arguments to update the paths.

Parameters:

  • model_args (dict) – The model arguments from where the new paths must be taken.

  • arch (Architecture) – The neural network architecture that must be updated, see Architecture.

add_to_state(state)

Add handled paths to state dictionary.

Parameters:

state (dict) – The state dictionary from a __getstate__() method. See simple_dl_model_handler.SimpleDLModelHandler.__getstate__() for an example.

Returns:

Nothing at all.

get_from_state(state)

Get paths to handle from state dictionary.

Parameters:

state (dict) – The state dictionary from a __setstate__() method. See simple_dl_model_handler.SimpleDLModelHandler.__setstate__() for an example.

Returns:

Nothing at all.