From 4da7a2c812221d56a430b35139ac40b23fa76f77 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 29 Jun 2021 22:54:52 +0200
Subject: Refactor of config, more granular

---
 training/conf/criterion/mse.yaml | 3 +++
 1 file changed, 3 insertions(+)
 create mode 100644 training/conf/criterion/mse.yaml

(limited to 'training/conf/criterion')

diff --git a/training/conf/criterion/mse.yaml b/training/conf/criterion/mse.yaml
new file mode 100644
index 0000000..4d89cbc
--- /dev/null
+++ b/training/conf/criterion/mse.yaml
@@ -0,0 +1,3 @@
+type: MSELoss
+args:
+  reduction: mean
-- 
cgit v1.2.3-70-g09d2