cellmap_segmentation_challenge.models.model_load

Functions

load_best_val(logs_save_path, ...[, low_is_best])	Load the model weights with the best validation score from a directory into an existing model object in place.
load_latest(search_path, model)	Load the latest checkpoint from a directory into a model (in place).

cellmap_segmentation_challenge.models.model_load.load_latest(search_path, model)

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)

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.