Spaces:
Sleeping
Sleeping
smhavens
commited on
Commit
·
5b3034b
1
Parent(s):
3fb98b7
Error fixing attempt 1
Browse files
app.py
CHANGED
@@ -82,6 +82,8 @@ def finetune(train_dataloader):
|
|
82 |
|
83 |
train_loss = losses.TripletLoss(model=model)
|
84 |
|
|
|
|
|
85 |
model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=10)
|
86 |
|
87 |
# accuracy = compute_metrics(eval, metric)
|
|
|
82 |
|
83 |
train_loss = losses.TripletLoss(model=model)
|
84 |
|
85 |
+
print(train_dataloader)
|
86 |
+
|
87 |
model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=10)
|
88 |
|
89 |
# accuracy = compute_metrics(eval, metric)
|