From 6fa34985737b7b36310d95dec7e9fb162a599a26 Mon Sep 17 00:00:00 2001 From: Jakub Pokrywka Date: Thu, 14 Jul 2022 19:19:12 +0000 Subject: [PATCH] hf roberta challam regression layer on top --- hf_roberta_base/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hf_roberta_base/config.py b/hf_roberta_base/config.py index 74a0ad1..61c7be3 100644 --- a/hf_roberta_base/config.py +++ b/hf_roberta_base/config.py @@ -1,4 +1,4 @@ -#MODEL = '/home/wmi/RoBERTa/without_date/checkpoint-1325000' -MODEL = 'roberta-base' +MODEL = '/mnt/gpu_data1/kubapok/RoBERTa/without_date/checkpoint-1325000' TEST=False +