From 3051db26297cac671994462aea9fcbd35e364103 Mon Sep 17 00:00:00 2001 From: Harshil <37377066+harshil21@users.noreply.github.com> Date: Sun, 2 Apr 2023 00:56:46 +0530 Subject: [PATCH] Update trust_and_safety_models/toxicity/train.py --- trust_and_safety_models/toxicity/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trust_and_safety_models/toxicity/train.py b/trust_and_safety_models/toxicity/train.py index 4e3a33659..3c0afe835 100644 --- a/trust_and_safety_models/toxicity/train.py +++ b/trust_and_safety_models/toxicity/train.py @@ -387,7 +387,7 @@ class Trainer(object): fold=i, ) else: - a = input("Are you sure you want to do multiple fold training? (y/n)") + a = input("Are you sure you want to do multiple fold training? (y/N)") if a.lower() == "y": for mb_generator, steps_per_epoch, val_data, test_data in self.mb_loader(full_df=df): self._train_single_fold(