fix loss info and typo (Closes GH-22).

2021-12-09 15:11:06 +08:00
parent 0d869161c2
commit 49cbc44069
16 changed files with 37 additions and 36 deletions
@@ -16,7 +16,7 @@
 * Loss function
 >  * Args
 >     * type: Loss function type, support `TripletLoss` and `CrossEntropyLoss`.
->     * loss_term_weights: loss weight.
+>     * loss_term_weight: loss weight.
 >     * log_prefix: the prefix of loss log.

 ----
@@ -107,11 +107,11 @@ evaluator_cfg:
      img_w: 64

 loss_cfg:
-  - loss_term_weights: 1.0
+  - loss_term_weight: 1.0
    margin: 0.2
    type: TripletLoss
    log_prefix: triplet
-  - loss_term_weights: 0.1
+  - loss_term_weight: 0.1
    scale: 16
    type: CrossEntropyLoss
    log_prefix: softmax