Update utils.py

ZhaoJ9014 · Mar 20, 2019 · b48f4f3 · b48f4f3
1 parent c0a416b
commit b48f4f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backup/utils.py b/backup/utils.py
@@ -167,7 +167,7 @@ def perform_val(multi_gpu, device, embedding_size, batch_size, backbone, carray,
     embeddings = np.zeros([len(carray), embedding_size])
     with torch.no_grad():
         while idx + batch_size <= len(carray):
-            batch = torch.tensor(carray[idx:idx + batch_size])
+            batch = torch.tensor(carray[idx:idx + batch_size][:, [2, 1, 0], :, :])
             if tta:
                 fliped = hflip_batch(batch)
                 emb_batch = backbone(batch.to(device)).cpu() + backbone(fliped.to(device)).cpu()
@@ -234,4 +234,4 @@ def accuracy(output, target, topk=(1,)):
 
 
 def save_checkpoint(state, filename):
-    torch.save(state, filename)
+    torch.save(state, filename)