INNER CODE UNIT · Python

main

zjunlp/PromptKG · lambdaKG/main.py:67

def main():
    parser = _setup_parser()
    args = parser.parse_args()
    args.gpus = torch.cuda.device_count()
    np.random.seed(args.seed)
    torch.manual_seed(args.seed)
    pl.seed_everything(args.seed)

    data_class = _import_class(f"data.{args.data_class}")
    litmodel_class = _import_class(f"lit_models.{args.lit_model_class}")
    

    # perfered , warp the transformers encoder
    method_name = args.model_class.lower().replace("model","")
    if method_name in metric_list:
        metric_name = metric_list[method_name]
    else:
        metric_name = "hits10"

View source record →