make it easier to do multiple fold training

trust_and_safety_models/toxicity/train.py

@@ -387,7 +387,8 @@ class Trainer(object):
         fold=i,
       )
     else:
-      raise ValueError("Sure you want to do multiple fold training")
+      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(
             mb_generator=mb_generator,