From 568ffd9205b9ecd7e4df6a4c6df851d3eeb8fbd6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ga=C5=A1per=20Spagnolo?= Date: Wed, 2 Aug 2023 16:24:44 +0200 Subject: [PATCH] Minori changes --- ft.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ft.py b/ft.py index 1a7a8fe..6fc32c2 100644 --- a/ft.py +++ b/ft.py @@ -129,10 +129,10 @@ class FT: def train(self): train_dataloader, test_dataloader = load_dataset( - "./dataset/hason_out.json", + "./dataset/hason_three.json", 200, 200, - 1, + 3, test_ratio=0.2, ) num_epochs = 100