From 393938d7f44044369cdd951306377d33fd4fb033 Mon Sep 17 00:00:00 2001 From: Utsav Dave Date: Tue, 4 Apr 2023 08:40:49 -0400 Subject: [PATCH] Updated trainer_utils.py --- twml/twml/contrib/trainers/trainer_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/twml/twml/contrib/trainers/trainer_utils.py b/twml/twml/contrib/trainers/trainer_utils.py index 5d71a7422..f279571be 100644 --- a/twml/twml/contrib/trainers/trainer_utils.py +++ b/twml/twml/contrib/trainers/trainer_utils.py @@ -56,7 +56,7 @@ def build_keras_trainer( metrics_fn: metrics function used by TensorFlow estimators. Defaults to twml.metrics.get_binary_class_metric_fn(). - **kwargs: for people familar with twml Trainer's options, they can be passed in here + **kwargs: for people familiar with twml Trainer's options, they can be passed in here as kwargs, and they will be forwarded to Trainer as opts. See https://cgit.twitter.biz/source/tree/twml/twml/argument_parser.py#n43 for available args.