downstream_classification.py

if __name__ == "__main__":
    from mylib.modeling.dialogue.baseline_dialogue_encoder import BaselineDialogueEncoderConfig
    from mylib.trainers.downstream_classification import get_configs, get_loaders, get_baseline_model, get_learner, train_or_val

    args, model_config, learner_config, trainer_config = get_configs(BaselineDialogueEncoderConfig)
    train_loader, val_loader = get_loaders(args, learner_config, trainer_config)
    model = get_baseline_model(model_config, learner_config)
    learner = get_learner(model, learner_config, trainer_config)
    train_or_val(args, learner, train_loader, val_loader, trainer_config)