smhavens commited on
Commit
5b3034b
·
1 Parent(s): 3fb98b7

Error fixing attempt 1

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)