how to load those saved models to continue training

marshuang80 / gloria

GLoRIA: A Multimodal Global-Local Representation Learning Framework forLabel-efficient Medical Image Recognition

Apache License 2.0

173 stars 29 forks source link

how to load those saved models to continue training #10

Closed church-XP closed 2 years ago

church-XP commented 2 years ago

I'm sorry to bother you again

I want to know how do I load those saved models ./gloria/data/ckpt/pneumothorax_segmentation_0.01/2022_08_03_12_23_46/epoch=35-step=143.ckpt

What is the difference between this and the pre-trained model How can I continue my previous training after the training is completed

I have tried replacing the pre-training model directly with epoch=35-step= 143.CKpt But I found a difference in the size of the two, and it was not successful,

marshuang80 commented 2 years ago

Hi @church-XP, please see examples for loading saved models here, here and here.

As the name suggests, the ckpt file you are referencing to is for segmentation.