Skip to content
Snippets Groups Projects
Commit 6d766af4 authored by mm2320's avatar mm2320
Browse files

Change train model to DeBERTa

parent 98135df1
No related branches found
No related tags found
No related merge requests found
......@@ -25,7 +25,7 @@ def train(env: Env) -> None:
if __name__ == '__main__':
model_type: ModelType = ModelType.DISTILBERT
model_type: ModelType = ModelType.DEBERTA
# Load initial hyperparameters
with open("configs/hyperparams_optim_yitang_1.json", mode="r") as f:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment