UniTrain
UniTrain copied to clipboard
segmentation: add a function for loading base model weights for transfer learning