Merge f21014f into 78ce392

google · Jun 29, 2020 · ef6d0c7 · ef6d0c7
2 parents 78ce392 + f21014f
commit ef6d0c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trax/layers/metrics.py b/trax/layers/metrics.py
@@ -20,8 +20,8 @@
 
   - model output: Batch of predicted values (typically vectors).
   - targets: Batch of target values (e.g., categories or vectors).
-  - weights: Tensor that can assign different weights to different positions in.
-    the model output. One common use of weights is for masking -- assigning
+  - weights: Tensor that can assign different weights to different positions
+    in the model output. One common use of weights is for masking -- assigning
     weight 0 to positions that correspond to padding in the input so that they
     don't affect metrics.