Move early_training_checks to train function

bin/run-ldc93s1.py

@@ -2,8 +2,7 @@
 import os
 from import_ldc93s1 import _download_and_preprocess_data as download_ldc
 from coqui_stt_training.util.config import initialize_globals_from_args
-from coqui_stt_training.train import train, test, early_training_checks
-import tensorflow.compat.v1 as tfv1
+from coqui_stt_training.train import train, test
 
 # only one GPU for only one training sample
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
@@ -21,8 +20,6 @@ initialize_globals_from_args(
     epochs=200,
 )
 
-early_training_checks()
-
 train()
-tfv1.reset_default_graph()
+
 test()

notebooks/easy-transfer-learning.ipynb

@@ -195,13 +195,11 @@
    },
    "outputs": [],
    "source": [
-    "from coqui_stt_training.train import train, early_training_checks\n",
+    "from coqui_stt_training.train import train\n",
     "\n",
     "# use maximum one GPU\n",
     "os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\"\n",
     "\n",
-    "early_training_checks()\n",
-    "\n",
     "train()"
    ]
   },

notebooks/train-your-first-coqui-STT-model.ipynb

@@ -197,13 +197,11 @@
    },
    "outputs": [],
    "source": [
-    "from coqui_stt_training.train import train, early_training_checks\n",
+    "from coqui_stt_training.train import train\n",
     "\n",
     "# use maximum one GPU\n",
     "os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\"\n",
     "\n",
-    "early_training_checks()\n",
-    "\n",
     "train()"
    ]
   },

training/coqui_stt_training/train.py

@@ -522,6 +522,8 @@ def log_grads_and_vars(grads_and_vars):
 
 
 def train():
+    early_training_checks()
+
     tfv1.reset_default_graph()
     tfv1.set_random_seed(Config.random_seed)
 
@@ -1266,7 +1268,6 @@ def early_training_checks():
 
 def main():
     initialize_globals_from_cli()
-    early_training_checks()
 
     if Config.train_files:
         train()