Commit 0df15d91 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

cleaning

parent 19d860e6
...@@ -48,14 +48,7 @@ from .preprocessor import RawPreprocessor ...@@ -48,14 +48,7 @@ from .preprocessor import RawPreprocessor
from .preprocessor import MfccFrontEnd from .preprocessor import MfccFrontEnd
from .preprocessor import MelSpecFrontEnd 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" __author__ = "Anthony Larcher and Sylvain Meignier"
......
...@@ -1492,17 +1492,17 @@ def xtrain(dataset_description, ...@@ -1492,17 +1492,17 @@ def xtrain(dataset_description,
validation_non_indices, validation_non_indices,
training_opts["mixed_precision"]) training_opts["mixed_precision"])
test_eer = None #test_eer = None
if training_opts["compute_test_eer"] and local_rank < 1: #if training_opts["compute_test_eer"] and local_rank < 1:
test_eer = test_metrics(model, device, model_opts, dataset_opts, training_opts) # test_eer = test_metrics(model, device, model_opts, dataset_opts, training_opts)
monitor.update(test_eer=test_eer, monitor.update(test_eer=test_eer,
val_eer=val_eer, val_eer=val_eer,
val_loss=val_loss, val_loss=val_loss,
val_acc=val_acc) val_acc=val_acc)
if local_rank < 1: #if local_rank < 1:
monitor.display() # monitor.display()
# Save the current model and if needed update the best one # 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 # TODO ajouter une option qui garde les modèles à certaines époques (par exemple avant le changement de LR
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment