Update metrics

2021-06-11 07:20:06 -07:00
parent 248686820c
commit 437f62b055
5 changed files with 158 additions and 1 deletions
--- a/configs/yaml.loss/top1-ce
+++ b/configs/yaml.loss/top1-ce
@@ -0,0 +1,12 @@
+class_or_func: ComposeMetric
+module_path: xautodl.xmisc.meter_utils
+args:
+  - class_or_func: Top1AccMetric
+    module_path: xautodl.xmisc.meter_utils
+    args: [False]
+    kwargs: {}
+  - class_or_func: CrossEntropyMetric
+    module_path: xautodl.xmisc.meter_utils
+    args: [False]
+    kwargs: {}
+kwargs: {}
--- a/configs/yaml.loss/top1.acc
+++ b/configs/yaml.loss/top1.acc
@@ -0,0 +1,4 @@
+class_or_func: Top1AccMetric
+module_path: xautodl.xmisc.meter_utils
+args: [False]
+kwargs: {}