From 783b011e06e75012eab3225e0755372094858ed9 Mon Sep 17 00:00:00 2001 From: gigasquid Date: Fri, 26 Apr 2019 10:14:53 -0400 Subject: [PATCH] fix docstring --- .../examples/bert/src/bert/bert_sentence_classification.clj | 4 ++-- 1 file changed, 2 insertions(+), 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 index ec9e0f693b6f..ea1720ebeb2c 100644 --- 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})