Minori changes
parent
557fac786d
commit
568ffd9205
4
ft.py
4
ft.py
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue