Finetuning text summary model support change pretrained model?

Amazon SageMaker
2

You can adjust the model, which should be used for training in the hyperparameters. Replace it with the model you want to use.


# hyperparameters, which are passed into the training job
hyperparameters={'per_device_train_batch_size': 4,
                 'per_device_eval_batch_size': 4,
                 'model_name_or_path': 'facebook/bart-large-cnn', # model used for training