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

cleaning

parent 19d860e6
......@@ -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"
......
......@@ -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
......
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