Spaces:
Sleeping
Sleeping
smhavens
commited on
Commit
·
4dec182
1
Parent(s):
20042af
Print for testing
Browse files
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 |
|