cellmap_segmentation_challenge.models.model_load#
Functions
|
Load the model weights with the best validation score from a directory into an existing model object in place. |
|
Load the latest checkpoint from a directory into a model (in place). |
- cellmap_segmentation_challenge.models.model_load.load_latest(search_path, model)[source]#
Load the latest checkpoint from a directory into a model (in place).
- Parameters:
search_path (str) – The path to search for checkpoints.
model (torch.nn.Module) – The model to load the checkpoint into.
- cellmap_segmentation_challenge.models.model_load.load_best_val(logs_save_path, model_save_path, model, low_is_best=True)[source]#
Load the model weights with the best validation score from a directory into an existing model object in place.
- Parameters:
logs_save_path (str) – The path to the directory with the tensorboard logs.
model_save_path (str) – The path to the model checkpoints.
model (torch.nn.Module) – The model to load the checkpoint into.
low_is_best (bool) – Whether a lower validation score is better.