speechbrain / benchmarks

This repository contains the SpeechBrain Benchmarks
Apache License 2.0
83 stars 35 forks source link

Fix Evaluation Logging in MOABB #27

Closed Drew-Wagner closed 5 months ago

Drew-Wagner commented 5 months ago

Fixes #24

After:

speechbrain.utils.checkpoints - Saved an end-of-epoch checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+02
speechbrain.utils.epoch_loop - Going into epoch 10
speechbrain.nnet.schedulers - Changing lr from 8.3e-06 to 9.7e-06
speechbrain.utils.train_logger - epoch: 10, lr: 6.95e-06 - train loss: 8.20e-01 - valid loss: 1.39, valid f1: 1.26e-01, valid acc: 2.50e-01, valid cm: [[13  1  0  0]
 [13  1  0  0]
 [13  1  0  0]
 [14  0  0  0]]
speechbrain.utils.checkpoints - Saved an end-of-epoch checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+03
speechbrain.utils.checkpoints - Loading a checkpoint from results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+03
speechbrain.utils.train_logger - epoch loaded: 10 - test loss: 1.39, test f1: 1.58e-01, test acc: 2.67e-01, test cm: [[65  7  0  0]
 [60 12  0  0]
 [67  5  0  0]
 [68  4  0  0]]
speechbrain.utils.checkpoints - Saved an end-of-epoch checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-45+00
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+02
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-43+00
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+00
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-43+03
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-43+02
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-42+01
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+01
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-43+01
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-42+00
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-44+03
speechbrain.utils.checkpoints - Deleted checkpoint in results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-45-41+03
speechbrain.utils.checkpoints - Loading a checkpoint from results/MotorImagery/BNCI2014001/EEGNet_train/leave-one-session-out/sub-001/1test/save/CKPT+2024-03-23+09-48-45+00
speechbrain.utils.train_logger - epoch loaded: 10 - valid loss: 2.52e-01, valid f1: 1.11e-01, valid acc: 2.14e-01, valid cm: [[11  3  0  0]
 [13  1  0  0]
 [13  1  0  0]
 [14  0  0  0]]
mravanelli commented 5 months ago

Thank you @Drew-Wagner for the fix. I tested it and everything works well.