smhavens commited on
Commit
4dec182
·
1 Parent(s): 20042af

Print for testing

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -139,6 +139,8 @@ def training():
139
 
140
  train_dataloader = DataLoader(train_examples, shuffle=True, batch_size=25)
141
 
 
 
142
  # print(train_examples)
143
 
144
  embeddings = finetune(train_dataloader)
@@ -158,6 +160,8 @@ def finetune(train_dataloader):
158
 
159
  train_loss = losses.BatchHardSoftMarginTripletLoss(model=model)
160
 
 
 
161
  model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=10)
162
 
163
 
 
139
 
140
  train_dataloader = DataLoader(train_examples, shuffle=True, batch_size=25)
141
 
142
+ print("END DATALOADER")
143
+
144
  # print(train_examples)
145
 
146
  embeddings = finetune(train_dataloader)
 
160
 
161
  train_loss = losses.BatchHardSoftMarginTripletLoss(model=model)
162
 
163
+ print("BEGIN FIT")
164
+
165
  model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=10)
166
 
167