Light cnn

bob/learn/pytorch/datasets/utils.py

@@ -90,7 +90,8 @@ def map_labels(raw_labels, start_index=0):
 
   for i in range(len(possible_labels)):
     l = possible_labels[i]
-    labels[numpy.where(labels==l)[0][0]] = i + start_index
+    #labels[numpy.where(labels==l)[0][0]] = i + start_index
+    labels[numpy.where(labels==l)[0]] = i + start_index
 
   # -----
   # map back to native int, resolve the problem with dataset concatenation