Closed yongzx closed 2 years ago
Further commits:
e1079c1 - use correct Trainer class, and use model.freeze_model(freeze=True)
to freeze the base model instead of using for-loop to freeze base model. The reason is that the latter would break if we use BERT model that has "bert" prefix instead of "transformer" prefix.
parser.add_argument("--task_layers", choices=task_layers, required=True)
.