spanskiduh
/
fine-tune
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Minori changes

main
Gašper Spagnolo 2023-08-02 16:24:44 +02:00
parent 557fac786d
commit 568ffd9205
No known key found for this signature in database
GPG Key ID: 2EA0738CC1EFEEB7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ft.py
View File

@ -129,10 +129,10 @@ class FT:
def train(self): def train(self):
train_dataloader, test_dataloader = load_dataset( train_dataloader, test_dataloader = load_dataset(
"./dataset/hason_out.json", "./dataset/hason_three.json",
200, 200,
200, 200,
1, 3,
test_ratio=0.2, test_ratio=0.2,
) )
num_epochs = 100 num_epochs = 100