fix: set caffe smooth l1 loss threshold to 1

Huber loss has threshold set to 1
jolibrain · Sep 15, 2020 · 0e329f0 · 0e329f0
1 parent 4e52a0e
commit 0e329f0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/backends/caffe/caffelib.cc b/src/backends/caffe/caffelib.cc
@@ -395,8 +395,7 @@ namespace dd
                     else if (_loss == "L1")
                       {
                         lparam->set_type("SmoothL1Loss");
-                        lparam->mutable_smooth_l1_loss_param()->set_sigma(
-                            10.0);
+                        lparam->mutable_smooth_l1_loss_param()->set_sigma(1.0);
                       }
                     else if (_loss == "sigmoid")
                       {