fix docstring

apache · Apr 26, 2019 · 783b011 · 783b011
1 parent a3293f3
commit 783b011
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/contrib/clojure-package/examples/bert/src/bert/bert_sentence_classification.clj b/contrib/clojure-package/examples/bert/src/bert/bert_sentence_classification.clj
@@ -67,8 +67,8 @@
 
 (defn fine-tune-model
   "msymbol: the pretrained network symbol
-   arg-params: the argument parameters of the pretrained model
-   num-classes: the number of classes for the fine-tune datasets"
+   num-classes: the number of classes for the fine-tune datasets
+   dropout: The dropout rate amount"
   [msymbol {:keys [num-classes dropout]}]
   (as-> msymbol data
     (sym/dropout {:data data :p dropout})