import os
from padmet.utils.constants import PRETRAINED_MODEL_FOLDER
from fsl.train_metadataset import MetaLearningLightningModule

if __name__ == '__main__':
    model_path = os.path.join(PRETRAINED_MODEL_FOLDER, 'selfsup_ckpts_cnn20.ckpt')
    MetaLearningLightningModule.load_from_checkpoint(model_path)