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

cleaning

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