Commit 2aacbe26 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

cleaning

parent 14f1672e
...@@ -1484,18 +1484,18 @@ def xtrain(dataset_description, ...@@ -1484,18 +1484,18 @@ def xtrain(dataset_description,
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
#if local_rank < 1: if local_rank < 1:
# save_model(model, monitor, model_opts, training_opts, optimizer, scheduler, epoch) save_model(model, monitor, model_opts, training_opts, optimizer, scheduler, epoch)
for ii in range(torch.cuda.device_count()): for ii in range(torch.cuda.device_count()):
......
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