From 4af1d651da54cb4aaf89abb4b49d01f47d09fd52 Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Sun, 1 Sep 2024 16:08:50 +0200 Subject: [PATCH] icnreased epoch --- src/model/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model/train.py b/src/model/train.py index 7f0671c..3b39b89 100644 --- a/src/model/train.py +++ b/src/model/train.py @@ -67,7 +67,7 @@ class FakeNewsModelTrainer: return TensorDataset(input_ids, attention_mask, labels), sampler - def train(self, train_data, val_data, epochs=5, batch_size=32): + def train(self, train_data, val_data, epochs=13, batch_size=16): train_dataset, train_sampler = train_data train_dataloader = DataLoader(train_dataset, sampler=train_sampler, batch_size=batch_size) val_dataloader = DataLoader(val_data, batch_size=batch_size, shuffle=False)