def get_model(model):
    "Return the model maybe wrapped inside `model`."    
    return model.module if isinstance(model, (DistributedDataParallel, nn.DataParallel)) else model
Return the model maybe wrapped inside `model`.
PatchTST_self_supervised/src/learner.py