Updated for logger.

sraashis · Dec 5, 2023 · dd0e0e2 · dd0e0e2
1 parent 49b7865
commit dd0e0e2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -59,7 +59,7 @@ class MNISTTrainer(ETRunner):
 
     def new_meter(self):
         return ETMeter(
-            cfm=ConfusionMatrix(num_classes=10)
+            cfm=ConfusionMatrix(num_classes=10, device=self.device['gpu'])
         )
 
 

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 # This call to setup() does all the work
 setup(
     name="easytorch",
-    version="3.8.3",
+    version="3.8.4",
     description="Easy Neural Network Experiments with pytorch",
     long_description=_README,
     long_description_content_type="text/markdown",