sfaira.train.GridsearchContainer.save_best_weight

GridsearchContainer.save_best_weight(path: str, partition: str = 'val', metric: str = 'loss', subset: dict = {})

Copies weight file from best hyperparameter setting from grid search directory to zoo directory with cleaned file name.

Parameters
  • path – Target file to save to. This is intended to be the zoo directory ready for upload.

  • partition

  • metric

  • subset

Returns