Fingerspelling embedding + ClearML

2023-05-21 20:30:12 +00:00
parent 2cbf11eb00
commit bd2b848eac
26 changed files with 2465 additions and 176 deletions
--- a/models/utils.py
+++ b/models/utils.py
@@ -88,9 +88,10 @@ def train_epoch_embedding_online(model, epoch_iters, train_loader, val_loader, c
        if enable_batch_sorting:
            if labels_size < train_loader.batch_size:
                trim_count = labels_size % mini_batch
-                inputs = inputs[:-trim_count]
-                labels = labels[:-trim_count]
-                masks = masks[:-trim_count]
+                if trim_count > 0:
+                    inputs = inputs[:-trim_count]
+                    labels = labels[:-trim_count]
+                    masks = masks[:-trim_count]
            embeddings = None
            with torch.no_grad():
                for j in range(batch_loop_count):