diff --git a/mtl_config.yaml b/mtl_config.yaml index 9085350282f585433e3536a292b0c4b78fd63f1f..e6bbcf66ff24803fc6e564171db4738b43240726 100644 --- a/mtl_config.yaml +++ b/mtl_config.yaml @@ -4,7 +4,7 @@ auxiliary_task: "mask_language_model answer_matching" do_train: True do_predict: True -use_cuda: False +use_cuda: True checkpoint_path: "output_model/firstrun"