cleaning

nnet/__init__.py

@@ -48,14 +48,7 @@ from .preprocessor import RawPreprocessor
 from .preprocessor import MfccFrontEnd 
 from .preprocessor import MelSpecFrontEnd
 
-has_pyroom = True
-try:
-   import pyroomacoustics
-except ImportError:
-   has_pyroom = False
 
-if has_pyroom:
-    from .augmentation import AddReverb
 
 
 __author__ = "Anthony Larcher and Sylvain Meignier"

nnet/xvector.py

@@ -1492,17 +1492,17 @@ def xtrain(dataset_description,
                                                           validation_non_indices,
                                                           training_opts["mixed_precision"])
 
-            test_eer = None
-            if training_opts["compute_test_eer"] and local_rank < 1:
-                test_eer = test_metrics(model, device, model_opts, dataset_opts, training_opts)
+            #test_eer = None
+            #if training_opts["compute_test_eer"] and local_rank < 1:
+            #    test_eer = test_metrics(model, device, model_opts, dataset_opts, training_opts)
 
             monitor.update(test_eer=test_eer,
                            val_eer=val_eer,
                            val_loss=val_loss,
                            val_acc=val_acc)
 
-            if local_rank < 1:
-                monitor.display()
+            #if local_rank < 1:
+            #    monitor.display()
 
             # Save the current model and if needed update the best one
             # TODO ajouter une option qui garde les modèles à certaines époques (par exemple avant le changement de LR