Spaces:
Running
Running
GitHub Actions commited on
Commit ·
d4c6bba
1
Parent(s): 91e3338
Sync from GitHub Actions
Browse files
visual_product_search/embeddings/train.py
CHANGED
|
@@ -80,6 +80,8 @@ def train(model, dataloader, device, epochs=5, lr=1e-5, grad_accum_steps=1, num_
|
|
| 80 |
raise ExceptionHandle(e, sys)
|
| 81 |
|
| 82 |
|
|
|
|
|
|
|
| 83 |
# for epoch in range(EPOCHS):
|
| 84 |
# model.train()
|
| 85 |
# total_loss = 0.0
|
|
|
|
| 80 |
raise ExceptionHandle(e, sys)
|
| 81 |
|
| 82 |
|
| 83 |
+
|
| 84 |
+
|
| 85 |
# for epoch in range(EPOCHS):
|
| 86 |
# model.train()
|
| 87 |
# total_loss = 0.0
|