Berufsschule/VeracityAI
Berufsschule
/
VeracityAI
Archived
3
0
Fork 0
Code Issues 9 Pull Requests Packages Projects Releases Wiki Activity

icnreased epoch

This commit is contained in:
Falko Victor Habel 2024-09-01 16:08:50 +02:00
parent da78ad357a
commit 4af1d651da
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model/train.py
View File

@ -67,7 +67,7 @@ class FakeNewsModelTrainer:
return TensorDataset(input_ids, attention_mask, labels), sampler 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_dataset, train_sampler = train_data
train_dataloader = DataLoader(train_dataset, sampler=train_sampler, batch_size=batch_size) train_dataloader = DataLoader(train_dataset, sampler=train_sampler, batch_size=batch_size)
val_dataloader = DataLoader(val_data, batch_size=batch_size, shuffle=False) val_dataloader = DataLoader(val_data, batch_size=batch_size, shuffle=False)