Update tools.py

tools.py

@@ -96,7 +96,7 @@ class evaluate_consumption_example(Tool):
         i=0
         for data in iter(test_dataset):
             print(i)
-            if (i<=50):
+            if (i<=5):
                 with torch.no_grad():
                     result = classifier(np.asarray(data["input_values"]), batch_size=64)
                 predicted_label = result[0]['label']
@@ -109,7 +109,7 @@ class evaluate_consumption_example(Tool):
                 torch.cuda.empty_cache()
                 gc.collect()
                 i=i+1
-            if(i>50):
+            if(i>5):
                 break
         logging.info("Fin des prédictions")
         del classifier