Upload SSL Layer MOS checkpoints (part 10)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_bvcc/w2v2_large_xlsr53/layer20/run2/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run2/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run2/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run3/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run3/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run3/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run3/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run4/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run4/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run4/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer20/run4/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run0/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run0/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run0/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run0/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run1/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run1/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run1/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run1/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run2/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run2/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run2/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run2/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run3/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run3/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run3/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run3/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run4/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run4/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run4/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer22/run4/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run0/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run0/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run0/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run0/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run1/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run1/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run1/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run1/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run2/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run2/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run2/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run2/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run3/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run3/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run3/model_best.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run3/train.log +99 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run4/config.gin +31 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run4/model.pt +3 -0
- results_bvcc/w2v2_large_xlsr53/layer4/run4/model_best.pt +3 -0
results_bvcc/w2v2_large_xlsr53/layer20/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5d14589be817c0d74382b4c45df57ba4f96a2d9a394b0defdfa20c7967059784
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer20/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31e283b49ce56846e5ea60201716f2775e95f09dbc524bb890f76e317df5e738
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_xlsr53/layer20/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5037704167457727
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3710 | LCC = 0.8137 | SRCC = 0.8130 ]
|
| 8 |
+
INFO:root:Average loss=0.3218344642947882
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3017 | LCC = 0.8344 | SRCC = 0.8325 ]
|
| 11 |
+
INFO:root:Average loss=0.28683051887231
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3121 | LCC = 0.8354 | SRCC = 0.8332 ]
|
| 14 |
+
INFO:root:Average loss=0.2438610845651382
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3209 | LCC = 0.8407 | SRCC = 0.8401 ]
|
| 17 |
+
INFO:root:Average loss=0.24924463224716675
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2590 | LCC = 0.8458 | SRCC = 0.8458 ]
|
| 20 |
+
INFO:root:Average loss=0.2369476317977294
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2576 | LCC = 0.8437 | SRCC = 0.8408 ]
|
| 23 |
+
INFO:root:Average loss=0.22474074516540918
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2660 | LCC = 0.8524 | SRCC = 0.8500 ]
|
| 26 |
+
INFO:root:Average loss=0.2444543399107762
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2875 | LCC = 0.8515 | SRCC = 0.8516 ]
|
| 29 |
+
INFO:root:Average loss=0.20904724892133322
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2772 | LCC = 0.8513 | SRCC = 0.8500 ]
|
| 32 |
+
INFO:root:Average loss=0.22854098400626427
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2458 | LCC = 0.8576 | SRCC = 0.8539 ]
|
| 35 |
+
INFO:root:Average loss=0.20622509364516306
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2966 | LCC = 0.8542 | SRCC = 0.8541 ]
|
| 38 |
+
INFO:root:Average loss=0.24844215485530022
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.4012 | LCC = 0.8432 | SRCC = 0.8474 ]
|
| 41 |
+
INFO:root:Average loss=0.23113030615525368
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2578 | LCC = 0.8575 | SRCC = 0.8540 ]
|
| 44 |
+
INFO:root:Average loss=0.20823164876454917
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2423 | LCC = 0.8570 | SRCC = 0.8573 ]
|
| 47 |
+
INFO:root:Average loss=0.19476968174179396
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2662 | LCC = 0.8567 | SRCC = 0.8562 ]
|
| 50 |
+
INFO:root:Average loss=0.20127781767111558
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2434 | LCC = 0.8608 | SRCC = 0.8588 ]
|
| 53 |
+
INFO:root:Average loss=0.1766820180301483
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2376 | LCC = 0.8623 | SRCC = 0.8589 ]
|
| 56 |
+
INFO:root:Average loss=0.17697289051153722
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2839 | LCC = 0.8550 | SRCC = 0.8540 ]
|
| 59 |
+
INFO:root:Average loss=0.17354973529775938
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2558 | LCC = 0.8608 | SRCC = 0.8563 ]
|
| 62 |
+
INFO:root:Average loss=0.17916993128183561
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2381 | LCC = 0.8615 | SRCC = 0.8606 ]
|
| 65 |
+
INFO:root:Average loss=0.17042449068946716
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3470 | LCC = 0.8488 | SRCC = 0.8525 ]
|
| 68 |
+
INFO:root:Average loss=0.1760022763449412
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2639 | LCC = 0.8549 | SRCC = 0.8551 ]
|
| 71 |
+
INFO:root:Average loss=0.16853642702484742
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2303 | LCC = 0.8604 | SRCC = 0.8592 ]
|
| 74 |
+
INFO:root:Average loss=0.15448664902494505
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2323 | LCC = 0.8632 | SRCC = 0.8597 ]
|
| 77 |
+
INFO:root:Average loss=0.15746389988523263
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3181 | LCC = 0.8536 | SRCC = 0.8569 ]
|
| 80 |
+
INFO:root:Average loss=0.15692691256602606
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2275 | LCC = 0.8571 | SRCC = 0.8586 ]
|
| 83 |
+
INFO:root:Average loss=0.14679662816417524
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2730 | LCC = 0.8571 | SRCC = 0.8559 ]
|
| 86 |
+
INFO:root:Average loss=0.13968717698485422
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2873 | LCC = 0.8537 | SRCC = 0.8529 ]
|
| 89 |
+
INFO:root:Average loss=0.14161112551123667
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2696 | LCC = 0.8556 | SRCC = 0.8553 ]
|
| 92 |
+
INFO:root:Average loss=0.13464587267774802
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2552 | LCC = 0.8610 | SRCC = 0.8577 ]
|
| 95 |
+
INFO:root:Average loss=0.134915314328212
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2346 | LCC = 0.8566 | SRCC = 0.8554 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2229 | LCC = 0.8692 | SRCC = 0.8727 ]
|
results_bvcc/w2v2_large_xlsr53/layer20/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer20/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7b5850ead9cca3ebc32292cd021f6c60bc35c88612ac958069def7a13b4be03
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer20/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:48420bd691f9a26d2a20781ee333fe6be696d58cbe7c68542d5149ad2b63fc7c
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_xlsr53/layer20/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5425823359535291
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3698 | LCC = 0.8129 | SRCC = 0.8186 ]
|
| 8 |
+
INFO:root:Average loss=0.32132799713275373
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3541 | LCC = 0.8196 | SRCC = 0.8255 ]
|
| 11 |
+
INFO:root:Average loss=0.2719362173707057
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3146 | LCC = 0.8396 | SRCC = 0.8391 ]
|
| 14 |
+
INFO:root:Average loss=0.2679524010954759
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3305 | LCC = 0.8328 | SRCC = 0.8417 ]
|
| 17 |
+
INFO:root:Average loss=0.25979573260515165
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2647 | LCC = 0.8423 | SRCC = 0.8431 ]
|
| 20 |
+
INFO:root:Average loss=0.2486117742000482
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2708 | LCC = 0.8522 | SRCC = 0.8505 ]
|
| 23 |
+
INFO:root:Average loss=0.24040728425368285
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2817 | LCC = 0.8404 | SRCC = 0.8411 ]
|
| 26 |
+
INFO:root:Average loss=0.22360487377796417
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2636 | LCC = 0.8546 | SRCC = 0.8516 ]
|
| 29 |
+
INFO:root:Average loss=0.2194032772229268
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2495 | LCC = 0.8537 | SRCC = 0.8512 ]
|
| 32 |
+
INFO:root:Average loss=0.2205862891024504
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2866 | LCC = 0.8585 | SRCC = 0.8552 ]
|
| 35 |
+
INFO:root:Average loss=0.20756436807987017
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2407 | LCC = 0.8547 | SRCC = 0.8531 ]
|
| 38 |
+
INFO:root:Average loss=0.2210563491934385
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2822 | LCC = 0.8585 | SRCC = 0.8546 ]
|
| 41 |
+
INFO:root:Average loss=0.20477527876694998
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2332 | LCC = 0.8603 | SRCC = 0.8590 ]
|
| 44 |
+
INFO:root:Average loss=0.19850699737285957
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2565 | LCC = 0.8544 | SRCC = 0.8524 ]
|
| 47 |
+
INFO:root:Average loss=0.1866154365050487
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2960 | LCC = 0.8608 | SRCC = 0.8589 ]
|
| 50 |
+
INFO:root:Average loss=0.18801097056040397
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2596 | LCC = 0.8573 | SRCC = 0.8543 ]
|
| 53 |
+
INFO:root:Average loss=0.1865861190435214
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2432 | LCC = 0.8593 | SRCC = 0.8560 ]
|
| 56 |
+
INFO:root:Average loss=0.19375459047464225
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2574 | LCC = 0.8593 | SRCC = 0.8575 ]
|
| 59 |
+
INFO:root:Average loss=0.18797000516683626
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2607 | LCC = 0.8636 | SRCC = 0.8605 ]
|
| 62 |
+
INFO:root:Average loss=0.1724727220642261
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2477 | LCC = 0.8620 | SRCC = 0.8591 ]
|
| 65 |
+
INFO:root:Average loss=0.16185139960203415
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2719 | LCC = 0.8507 | SRCC = 0.8535 ]
|
| 68 |
+
INFO:root:Average loss=0.1838340662801877
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2447 | LCC = 0.8636 | SRCC = 0.8593 ]
|
| 71 |
+
INFO:root:Average loss=0.17008591290467825
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3163 | LCC = 0.8507 | SRCC = 0.8530 ]
|
| 74 |
+
INFO:root:Average loss=0.16352493315935135
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2311 | LCC = 0.8599 | SRCC = 0.8568 ]
|
| 77 |
+
INFO:root:Average loss=0.15196033337941536
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2397 | LCC = 0.8593 | SRCC = 0.8537 ]
|
| 80 |
+
INFO:root:Average loss=0.14883078491458526
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2508 | LCC = 0.8544 | SRCC = 0.8538 ]
|
| 83 |
+
INFO:root:Average loss=0.15664823410602716
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2321 | LCC = 0.8595 | SRCC = 0.8555 ]
|
| 86 |
+
INFO:root:Average loss=0.1461077043070243
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2510 | LCC = 0.8575 | SRCC = 0.8537 ]
|
| 89 |
+
INFO:root:Average loss=0.13786816090727463
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2476 | LCC = 0.8555 | SRCC = 0.8539 ]
|
| 92 |
+
INFO:root:Average loss=0.13456128919735932
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2449 | LCC = 0.8544 | SRCC = 0.8482 ]
|
| 95 |
+
INFO:root:Average loss=0.13000222629843614
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2385 | LCC = 0.8561 | SRCC = 0.8531 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2623 | LCC = 0.8731 | SRCC = 0.8742 ]
|
results_bvcc/w2v2_large_xlsr53/layer20/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer20/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c50488e9e8b9f83e3743a1e0aa60b31d7e3c38e45c421b06da3c734e2743a1f1
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer20/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f557d716907f5344d87c1ad00ae44e9c99276ded924802742444e4ad13ddeb2
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_xlsr53/layer20/run4/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5252037472449816
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3260 | LCC = 0.8098 | SRCC = 0.8132 ]
|
| 8 |
+
INFO:root:Average loss=0.3006973887483279
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2792 | LCC = 0.8349 | SRCC = 0.8337 ]
|
| 11 |
+
INFO:root:Average loss=0.28456209275202876
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2737 | LCC = 0.8432 | SRCC = 0.8399 ]
|
| 14 |
+
INFO:root:Average loss=0.2644761571517357
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2677 | LCC = 0.8417 | SRCC = 0.8395 ]
|
| 17 |
+
INFO:root:Average loss=0.2482989699794696
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2976 | LCC = 0.8481 | SRCC = 0.8462 ]
|
| 20 |
+
INFO:root:Average loss=0.23510840726204407
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2639 | LCC = 0.8520 | SRCC = 0.8486 ]
|
| 23 |
+
INFO:root:Average loss=0.26927776319476276
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2799 | LCC = 0.8532 | SRCC = 0.8506 ]
|
| 26 |
+
INFO:root:Average loss=0.22184919241147164
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2623 | LCC = 0.8512 | SRCC = 0.8503 ]
|
| 29 |
+
INFO:root:Average loss=0.22584270418454439
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2872 | LCC = 0.8481 | SRCC = 0.8487 ]
|
| 32 |
+
INFO:root:Average loss=0.213427450412359
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2583 | LCC = 0.8548 | SRCC = 0.8506 ]
|
| 35 |
+
INFO:root:Average loss=0.20632134740933394
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2672 | LCC = 0.8518 | SRCC = 0.8513 ]
|
| 38 |
+
INFO:root:Average loss=0.20528878816045248
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2751 | LCC = 0.8532 | SRCC = 0.8495 ]
|
| 41 |
+
INFO:root:Average loss=0.20711010006757882
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2598 | LCC = 0.8545 | SRCC = 0.8502 ]
|
| 44 |
+
INFO:root:Average loss=0.20164973871448102
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2799 | LCC = 0.8524 | SRCC = 0.8528 ]
|
| 47 |
+
INFO:root:Average loss=0.1846661360409015
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2438 | LCC = 0.8565 | SRCC = 0.8528 ]
|
| 50 |
+
INFO:root:Average loss=0.18137922004247323
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3013 | LCC = 0.8507 | SRCC = 0.8487 ]
|
| 53 |
+
INFO:root:Average loss=0.19190689692130455
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2794 | LCC = 0.8563 | SRCC = 0.8547 ]
|
| 56 |
+
INFO:root:Average loss=0.18160534296662378
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2580 | LCC = 0.8580 | SRCC = 0.8546 ]
|
| 59 |
+
INFO:root:Average loss=0.16761686232609627
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2677 | LCC = 0.8524 | SRCC = 0.8545 ]
|
| 62 |
+
INFO:root:Average loss=0.17268649517343596
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2744 | LCC = 0.8553 | SRCC = 0.8557 ]
|
| 65 |
+
INFO:root:Average loss=0.16771187575963828
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2467 | LCC = 0.8589 | SRCC = 0.8576 ]
|
| 68 |
+
INFO:root:Average loss=0.16012380635127044
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2472 | LCC = 0.8602 | SRCC = 0.8587 ]
|
| 71 |
+
INFO:root:Average loss=0.1710203097989926
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2308 | LCC = 0.8568 | SRCC = 0.8537 ]
|
| 74 |
+
INFO:root:Average loss=0.15017850888081086
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2880 | LCC = 0.8546 | SRCC = 0.8551 ]
|
| 77 |
+
INFO:root:Average loss=0.14663071156694338
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2700 | LCC = 0.8556 | SRCC = 0.8555 ]
|
| 80 |
+
INFO:root:Average loss=0.14466455894020888
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2646 | LCC = 0.8584 | SRCC = 0.8580 ]
|
| 83 |
+
INFO:root:Average loss=0.1419999205913299
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2210 | LCC = 0.8589 | SRCC = 0.8565 ]
|
| 86 |
+
INFO:root:Average loss=0.14327592144791895
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2841 | LCC = 0.8573 | SRCC = 0.8544 ]
|
| 89 |
+
INFO:root:Average loss=0.13189556769644603
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2464 | LCC = 0.8553 | SRCC = 0.8540 ]
|
| 92 |
+
INFO:root:Average loss=0.12708164933018196
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2558 | LCC = 0.8581 | SRCC = 0.8559 ]
|
| 95 |
+
INFO:root:Average loss=0.1344688210922938
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2375 | LCC = 0.8540 | SRCC = 0.8516 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2276 | LCC = 0.8744 | SRCC = 0.8738 ]
|
results_bvcc/w2v2_large_xlsr53/layer22/run0/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer22/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d1a56f815795e931cc454630646df20dca752d5208210c5deb5516fcfe734a0c
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer22/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8e5f07ead89a4d410d80619c52e4a0485ab0e3c54493edec80831b864f3770d
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer22/run0/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.814239657842196
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.8270 | LCC = -0.0026 | SRCC = 0.0041 ]
|
| 8 |
+
INFO:root:Average loss=0.8085363232172452
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.8259 | LCC = 0.0493 | SRCC = 0.0578 ]
|
| 11 |
+
INFO:root:Average loss=0.8069201738406451
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.8247 | LCC = 0.1962 | SRCC = 0.1724 ]
|
| 14 |
+
INFO:root:Average loss=0.804259663208937
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.8243 | LCC = 0.1360 | SRCC = 0.0895 ]
|
| 17 |
+
INFO:root:Average loss=0.7963876479711288
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.8091 | LCC = 0.2472 | SRCC = 0.2498 ]
|
| 20 |
+
INFO:root:Average loss=0.7853928804397583
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.7979 | LCC = 0.2328 | SRCC = 0.2393 ]
|
| 23 |
+
INFO:root:Average loss=0.7715532069022839
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.7915 | LCC = 0.2385 | SRCC = 0.2449 ]
|
| 26 |
+
INFO:root:Average loss=0.7595658860145471
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.7796 | LCC = 0.2547 | SRCC = 0.2535 ]
|
| 29 |
+
INFO:root:Average loss=0.7552185222888604
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.7782 | LCC = 0.2615 | SRCC = 0.2638 ]
|
| 32 |
+
INFO:root:Average loss=0.7525569139382778
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.7804 | LCC = 0.2499 | SRCC = 0.2511 ]
|
| 35 |
+
INFO:root:Average loss=0.7490798051540668
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.7807 | LCC = 0.2506 | SRCC = 0.2477 ]
|
| 38 |
+
INFO:root:Average loss=0.7523076916352297
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.7763 | LCC = 0.2525 | SRCC = 0.2499 ]
|
| 41 |
+
INFO:root:Average loss=0.7473140374208108
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.7775 | LCC = 0.2565 | SRCC = 0.2576 ]
|
| 44 |
+
INFO:root:Average loss=0.7457434053604419
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.7724 | LCC = 0.2590 | SRCC = 0.2566 ]
|
| 47 |
+
INFO:root:Average loss=0.7413836671755865
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.7757 | LCC = 0.2561 | SRCC = 0.2580 ]
|
| 50 |
+
INFO:root:Average loss=0.746223298402933
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.7791 | LCC = 0.2531 | SRCC = 0.2559 ]
|
| 53 |
+
INFO:root:Average loss=0.7420797638404064
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.7749 | LCC = 0.2586 | SRCC = 0.2605 ]
|
| 56 |
+
INFO:root:Average loss=0.7429421719832298
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.7730 | LCC = 0.2617 | SRCC = 0.2634 ]
|
| 59 |
+
INFO:root:Average loss=0.7376499966933177
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.7722 | LCC = 0.2626 | SRCC = 0.2612 ]
|
| 62 |
+
INFO:root:Average loss=0.7371719991549467
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.7723 | LCC = 0.2641 | SRCC = 0.2609 ]
|
| 65 |
+
INFO:root:Average loss=0.7375246989421356
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.7737 | LCC = 0.2596 | SRCC = 0.2569 ]
|
| 68 |
+
INFO:root:Average loss=0.7357758390597808
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.7709 | LCC = 0.2627 | SRCC = 0.2593 ]
|
| 71 |
+
INFO:root:Average loss=0.7394503248043549
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.7721 | LCC = 0.2609 | SRCC = 0.2602 ]
|
| 74 |
+
INFO:root:Average loss=0.7379284046399288
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.7725 | LCC = 0.2575 | SRCC = 0.2581 ]
|
| 77 |
+
INFO:root:Average loss=0.7340310563643774
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.7736 | LCC = 0.2577 | SRCC = 0.2566 ]
|
| 80 |
+
INFO:root:Average loss=0.732508102288613
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.7724 | LCC = 0.2564 | SRCC = 0.2555 ]
|
| 83 |
+
INFO:root:Average loss=0.7312567891218723
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.7739 | LCC = 0.2568 | SRCC = 0.2540 ]
|
| 86 |
+
INFO:root:Average loss=0.7326350448987423
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.7737 | LCC = 0.2552 | SRCC = 0.2567 ]
|
| 89 |
+
INFO:root:Average loss=0.7306141421580926
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.7751 | LCC = 0.2552 | SRCC = 0.2572 ]
|
| 92 |
+
INFO:root:Average loss=0.7304448515940936
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.7763 | LCC = 0.2522 | SRCC = 0.2621 ]
|
| 95 |
+
INFO:root:Average loss=0.7258755343082623
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.7713 | LCC = 0.2600 | SRCC = 0.2612 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.8067 | LCC = 0.2564 | SRCC = 0.2638 ]
|
results_bvcc/w2v2_large_xlsr53/layer22/run1/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer22/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2a4207c575fda017ff98028b9abcd06aa3d9d89765d6e50c7f88d11e2c0f6f64
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer22/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:13807fed47bdbbf469ba629fdbafecfe255caee85de995c0a35f4000aaa2fed2
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer22/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.8181307224126962
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.8383 | LCC = 0.1263 | SRCC = 0.1136 ]
|
| 8 |
+
INFO:root:Average loss=0.8058509589769901
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.8313 | LCC = 0.1621 | SRCC = 0.1426 ]
|
| 11 |
+
INFO:root:Average loss=0.8050016615635309
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.8302 | LCC = 0.1852 | SRCC = 0.1710 ]
|
| 14 |
+
INFO:root:Average loss=0.8022013230201526
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.8238 | LCC = 0.2060 | SRCC = 0.2061 ]
|
| 17 |
+
INFO:root:Average loss=0.7932518109297141
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.8095 | LCC = 0.2203 | SRCC = 0.2173 ]
|
| 20 |
+
INFO:root:Average loss=0.7824825889024979
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.8238 | LCC = 0.2229 | SRCC = 0.2354 ]
|
| 23 |
+
INFO:root:Average loss=0.7676528906210874
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.7966 | LCC = 0.2300 | SRCC = 0.2333 ]
|
| 26 |
+
INFO:root:Average loss=0.7637531146025046
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.7890 | LCC = 0.2374 | SRCC = 0.2388 ]
|
| 29 |
+
INFO:root:Average loss=0.7614126335351895
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.7893 | LCC = 0.2364 | SRCC = 0.2401 ]
|
| 32 |
+
INFO:root:Average loss=0.7546045344609481
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.7927 | LCC = 0.2397 | SRCC = 0.2433 ]
|
| 35 |
+
INFO:root:Average loss=0.7521014488660372
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.7971 | LCC = 0.2430 | SRCC = 0.2452 ]
|
| 38 |
+
INFO:root:Average loss=0.7542945650907663
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.8035 | LCC = 0.2403 | SRCC = 0.2467 ]
|
| 41 |
+
INFO:root:Average loss=0.7496093186812524
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.8264 | LCC = 0.2363 | SRCC = 0.2470 ]
|
| 44 |
+
INFO:root:Average loss=0.7481262401128427
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.8146 | LCC = 0.2366 | SRCC = 0.2440 ]
|
| 47 |
+
INFO:root:Average loss=0.7483717394180787
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.7988 | LCC = 0.2392 | SRCC = 0.2478 ]
|
| 50 |
+
INFO:root:Average loss=0.7470266200028933
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.7773 | LCC = 0.2472 | SRCC = 0.2479 ]
|
| 53 |
+
INFO:root:Average loss=0.7472251134041028
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.8218 | LCC = 0.2336 | SRCC = 0.2444 ]
|
| 56 |
+
INFO:root:Average loss=0.7448329543456053
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.8404 | LCC = 0.2386 | SRCC = 0.2471 ]
|
| 59 |
+
INFO:root:Average loss=0.7468257385950822
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.8102 | LCC = 0.2413 | SRCC = 0.2495 ]
|
| 62 |
+
INFO:root:Average loss=0.7398762595959198
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.7956 | LCC = 0.2506 | SRCC = 0.2513 ]
|
| 65 |
+
INFO:root:Average loss=0.739950743623269
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.7798 | LCC = 0.2558 | SRCC = 0.2576 ]
|
| 68 |
+
INFO:root:Average loss=0.74148706977184
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.7825 | LCC = 0.2547 | SRCC = 0.2602 ]
|
| 71 |
+
INFO:root:Average loss=0.7412474002593603
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.8015 | LCC = 0.2487 | SRCC = 0.2586 ]
|
| 74 |
+
INFO:root:Average loss=0.7376955036933606
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.7965 | LCC = 0.2547 | SRCC = 0.2586 ]
|
| 77 |
+
INFO:root:Average loss=0.7389899431130825
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.8086 | LCC = 0.2474 | SRCC = 0.2556 ]
|
| 80 |
+
INFO:root:Average loss=0.7341073384651771
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.7827 | LCC = 0.2564 | SRCC = 0.2639 ]
|
| 83 |
+
INFO:root:Average loss=0.7389515500802261
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.7884 | LCC = 0.2528 | SRCC = 0.2571 ]
|
| 86 |
+
INFO:root:Average loss=0.7339698744890018
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.7711 | LCC = 0.2622 | SRCC = 0.2609 ]
|
| 89 |
+
INFO:root:Average loss=0.737748962182265
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.7770 | LCC = 0.2527 | SRCC = 0.2605 ]
|
| 92 |
+
INFO:root:Average loss=0.7384612575555459
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.7779 | LCC = 0.2590 | SRCC = 0.2643 ]
|
| 95 |
+
INFO:root:Average loss=0.7393181075652441
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.8223 | LCC = 0.2459 | SRCC = 0.2583 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.7996 | LCC = 0.2422 | SRCC = 0.2504 ]
|
results_bvcc/w2v2_large_xlsr53/layer22/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer22/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:600e13b24b4df5a9bad0859914464388871330115787498ebd9cf9ef13c194bc
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer22/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bc6b697fc358ef048a7d3407572dcc3c9f3ee6f595b36f5cc664485cc6f7138c
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_xlsr53/layer22/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.8170791971377838
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.8325 | LCC = 0.1026 | SRCC = 0.1114 ]
|
| 8 |
+
INFO:root:Average loss=0.8104762167502673
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.8263 | LCC = -0.0209 | SRCC = -0.0027 ]
|
| 11 |
+
INFO:root:Average loss=0.8071949627154913
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.8266 | LCC = 0.0121 | SRCC = 0.0360 ]
|
| 14 |
+
INFO:root:Average loss=0.8096001698420598
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.8256 | LCC = 0.0345 | SRCC = 0.0465 ]
|
| 17 |
+
INFO:root:Average loss=0.8068318619177892
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.8247 | LCC = 0.0963 | SRCC = 0.1003 ]
|
| 20 |
+
INFO:root:Average loss=0.8068208495775858
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.8289 | LCC = 0.1201 | SRCC = 0.1209 ]
|
| 23 |
+
INFO:root:Average loss=0.8082066774368286
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.8299 | LCC = 0.1215 | SRCC = 0.1197 ]
|
| 26 |
+
INFO:root:Average loss=0.7999909252692492
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.8185 | LCC = 0.2234 | SRCC = 0.2272 ]
|
| 29 |
+
INFO:root:Average loss=0.7974462914161193
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.8092 | LCC = 0.2301 | SRCC = 0.2401 ]
|
| 32 |
+
INFO:root:Average loss=0.7786302451904004
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.8018 | LCC = 0.2356 | SRCC = 0.2532 ]
|
| 35 |
+
INFO:root:Average loss=0.7637408131208175
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.8020 | LCC = 0.2301 | SRCC = 0.2432 ]
|
| 38 |
+
INFO:root:Average loss=0.7688980881984417
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.7847 | LCC = 0.2382 | SRCC = 0.2453 ]
|
| 41 |
+
INFO:root:Average loss=0.7564181311008258
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.7874 | LCC = 0.2371 | SRCC = 0.2423 ]
|
| 44 |
+
INFO:root:Average loss=0.7545683070635184
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.8091 | LCC = 0.2361 | SRCC = 0.2466 ]
|
| 47 |
+
INFO:root:Average loss=0.7512389161647894
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.7793 | LCC = 0.2450 | SRCC = 0.2449 ]
|
| 50 |
+
INFO:root:Average loss=0.7586557078055847
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.7864 | LCC = 0.2386 | SRCC = 0.2431 ]
|
| 53 |
+
INFO:root:Average loss=0.7566420504680047
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.7858 | LCC = 0.2423 | SRCC = 0.2426 ]
|
| 56 |
+
INFO:root:Average loss=0.7498466303715339
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.7781 | LCC = 0.2444 | SRCC = 0.2435 ]
|
| 59 |
+
INFO:root:Average loss=0.7462894381620945
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.8138 | LCC = 0.2368 | SRCC = 0.2465 ]
|
| 62 |
+
INFO:root:Average loss=0.7527802273249015
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.7938 | LCC = 0.2362 | SRCC = 0.2415 ]
|
| 65 |
+
INFO:root:Average loss=0.7473375663543359
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.7803 | LCC = 0.2437 | SRCC = 0.2459 ]
|
| 68 |
+
INFO:root:Average loss=0.7387752280785487
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.7824 | LCC = 0.2478 | SRCC = 0.2484 ]
|
| 71 |
+
INFO:root:Average loss=0.7506021536313571
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.7798 | LCC = 0.2467 | SRCC = 0.2469 ]
|
| 74 |
+
INFO:root:Average loss=0.742162606655023
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.7770 | LCC = 0.2476 | SRCC = 0.2459 ]
|
| 77 |
+
INFO:root:Average loss=0.7410742491483688
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.7751 | LCC = 0.2503 | SRCC = 0.2458 ]
|
| 80 |
+
INFO:root:Average loss=0.7445911574058044
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.7795 | LCC = 0.2432 | SRCC = 0.2463 ]
|
| 83 |
+
INFO:root:Average loss=0.7379241952529321
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.7901 | LCC = 0.2368 | SRCC = 0.2378 ]
|
| 86 |
+
INFO:root:Average loss=0.7360416750113169
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.7746 | LCC = 0.2503 | SRCC = 0.2480 ]
|
| 89 |
+
INFO:root:Average loss=0.7346521952213385
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.7962 | LCC = 0.2415 | SRCC = 0.2478 ]
|
| 92 |
+
INFO:root:Average loss=0.7360071792052343
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.7771 | LCC = 0.2491 | SRCC = 0.2513 ]
|
| 95 |
+
INFO:root:Average loss=0.7348292752718314
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.7890 | LCC = 0.2410 | SRCC = 0.2432 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.7907 | LCC = 0.2665 | SRCC = 0.2687 ]
|
results_bvcc/w2v2_large_xlsr53/layer22/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer22/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9d90223c11566e5b9366d1f191619706c6868e92086587ede254eea29795319c
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer22/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5055a81939fd491e91cffaa54a4a2efad6ceca33ada7ff52dff767a619970f25
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer22/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.8139951924482981
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.8538 | LCC = 0.0628 | SRCC = 0.0433 ]
|
| 8 |
+
INFO:root:Average loss=0.8084817964297074
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.8346 | LCC = 0.1495 | SRCC = 0.1408 ]
|
| 11 |
+
INFO:root:Average loss=0.8109443149505517
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.8399 | LCC = 0.1735 | SRCC = 0.1573 ]
|
| 14 |
+
INFO:root:Average loss=0.8042151973797724
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.8378 | LCC = 0.2081 | SRCC = 0.2045 ]
|
| 17 |
+
INFO:root:Average loss=0.8034162750610938
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.8440 | LCC = 0.2189 | SRCC = 0.2294 ]
|
| 20 |
+
INFO:root:Average loss=0.807106227446825
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.8303 | LCC = 0.2063 | SRCC = 0.2054 ]
|
| 23 |
+
INFO:root:Average loss=0.7967051993577908
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.8262 | LCC = 0.2212 | SRCC = 0.2231 ]
|
| 26 |
+
INFO:root:Average loss=0.7873269961430476
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.8369 | LCC = 0.2284 | SRCC = 0.2433 ]
|
| 29 |
+
INFO:root:Average loss=0.7755554616451263
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.8117 | LCC = 0.2389 | SRCC = 0.2573 ]
|
| 32 |
+
INFO:root:Average loss=0.7619453622744634
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.7796 | LCC = 0.2543 | SRCC = 0.2549 ]
|
| 35 |
+
INFO:root:Average loss=0.7669727420195555
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.7817 | LCC = 0.2582 | SRCC = 0.2596 ]
|
| 38 |
+
INFO:root:Average loss=0.7573283819051889
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.8115 | LCC = 0.2542 | SRCC = 0.2628 ]
|
| 41 |
+
INFO:root:Average loss=0.753503374564342
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.7973 | LCC = 0.2575 | SRCC = 0.2668 ]
|
| 44 |
+
INFO:root:Average loss=0.7453485207679944
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.8421 | LCC = 0.2580 | SRCC = 0.2705 ]
|
| 47 |
+
INFO:root:Average loss=0.7502139508724213
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.7675 | LCC = 0.2719 | SRCC = 0.2685 ]
|
| 50 |
+
INFO:root:Average loss=0.7436514829213803
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.7981 | LCC = 0.2540 | SRCC = 0.2611 ]
|
| 53 |
+
INFO:root:Average loss=0.7450853158266116
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.7642 | LCC = 0.2791 | SRCC = 0.2766 ]
|
| 56 |
+
INFO:root:Average loss=0.7439048061004052
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.7791 | LCC = 0.2748 | SRCC = 0.2763 ]
|
| 59 |
+
INFO:root:Average loss=0.7414584794105628
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.7793 | LCC = 0.2727 | SRCC = 0.2757 ]
|
| 62 |
+
INFO:root:Average loss=0.7410066070464941
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.7671 | LCC = 0.2795 | SRCC = 0.2782 ]
|
| 65 |
+
INFO:root:Average loss=0.7383616111981564
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.7835 | LCC = 0.2746 | SRCC = 0.2757 ]
|
| 68 |
+
INFO:root:Average loss=0.7370491738502796
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.8018 | LCC = 0.2653 | SRCC = 0.2673 ]
|
| 71 |
+
INFO:root:Average loss=0.7324779484516535
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.7692 | LCC = 0.2761 | SRCC = 0.2786 ]
|
| 74 |
+
INFO:root:Average loss=0.740232072579555
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.7750 | LCC = 0.2804 | SRCC = 0.2792 ]
|
| 77 |
+
INFO:root:Average loss=0.7392700532307992
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.7744 | LCC = 0.2807 | SRCC = 0.2760 ]
|
| 80 |
+
INFO:root:Average loss=0.7341036876806846
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.7650 | LCC = 0.2814 | SRCC = 0.2765 ]
|
| 83 |
+
INFO:root:Average loss=0.7312420346798041
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.7766 | LCC = 0.2829 | SRCC = 0.2840 ]
|
| 86 |
+
INFO:root:Average loss=0.7305280428666335
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.7855 | LCC = 0.2772 | SRCC = 0.2782 ]
|
| 89 |
+
INFO:root:Average loss=0.7277241028272189
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.7805 | LCC = 0.2800 | SRCC = 0.2774 ]
|
| 92 |
+
INFO:root:Average loss=0.7310245992281498
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.7570 | LCC = 0.2933 | SRCC = 0.2889 ]
|
| 95 |
+
INFO:root:Average loss=0.7278549052201785
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.7954 | LCC = 0.2802 | SRCC = 0.2838 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.7960 | LCC = 0.2681 | SRCC = 0.2799 ]
|
results_bvcc/w2v2_large_xlsr53/layer22/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer22/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e9bb001e0a7e711f1bb9275c034935d862ccdc20710f9458f8eefba1a22f9319
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer22/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:439a309d13eef5a0952916c0c3681ace23ef98e55fffd31d55e0902364494747
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer22/run4/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.8078734744817783
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.8438 | LCC = 0.1551 | SRCC = 0.1338 ]
|
| 8 |
+
INFO:root:Average loss=0.8061453310342935
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.8527 | LCC = 0.1858 | SRCC = 0.1696 ]
|
| 11 |
+
INFO:root:Average loss=0.7980850124970461
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.8782 | LCC = 0.2039 | SRCC = 0.2082 ]
|
| 14 |
+
INFO:root:Average loss=0.7904722224443387
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.8657 | LCC = 0.2205 | SRCC = 0.2218 ]
|
| 17 |
+
INFO:root:Average loss=0.7704049562796568
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.8042 | LCC = 0.2416 | SRCC = 0.2469 ]
|
| 20 |
+
INFO:root:Average loss=0.7644153917447115
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.8250 | LCC = 0.2385 | SRCC = 0.2437 ]
|
| 23 |
+
INFO:root:Average loss=0.7613810132711362
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.8584 | LCC = 0.2353 | SRCC = 0.2405 ]
|
| 26 |
+
INFO:root:Average loss=0.7635564146897732
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.8602 | LCC = 0.2334 | SRCC = 0.2399 ]
|
| 29 |
+
INFO:root:Average loss=0.7606048110203866
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.7999 | LCC = 0.2399 | SRCC = 0.2370 ]
|
| 32 |
+
INFO:root:Average loss=0.7518155513665615
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.7966 | LCC = 0.2453 | SRCC = 0.2419 ]
|
| 35 |
+
INFO:root:Average loss=0.7531586511012835
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.8018 | LCC = 0.2439 | SRCC = 0.2402 ]
|
| 38 |
+
INFO:root:Average loss=0.7529040429836664
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.8175 | LCC = 0.2530 | SRCC = 0.2528 ]
|
| 41 |
+
INFO:root:Average loss=0.7498894028174572
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.8310 | LCC = 0.2520 | SRCC = 0.2505 ]
|
| 44 |
+
INFO:root:Average loss=0.7508819340131222
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.8223 | LCC = 0.2518 | SRCC = 0.2502 ]
|
| 47 |
+
INFO:root:Average loss=0.7533572163337316
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.8383 | LCC = 0.2479 | SRCC = 0.2476 ]
|
| 50 |
+
INFO:root:Average loss=0.7443092568562582
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.8035 | LCC = 0.2544 | SRCC = 0.2475 ]
|
| 53 |
+
INFO:root:Average loss=0.7448368473694875
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.8218 | LCC = 0.2568 | SRCC = 0.2515 ]
|
| 56 |
+
INFO:root:Average loss=0.748010686574838
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.8373 | LCC = 0.2558 | SRCC = 0.2524 ]
|
| 59 |
+
INFO:root:Average loss=0.7410632696671363
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.8248 | LCC = 0.2561 | SRCC = 0.2532 ]
|
| 62 |
+
INFO:root:Average loss=0.7445670266946157
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.8269 | LCC = 0.2542 | SRCC = 0.2514 ]
|
| 65 |
+
INFO:root:Average loss=0.74035588823832
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.8644 | LCC = 0.2530 | SRCC = 0.2545 ]
|
| 68 |
+
INFO:root:Average loss=0.7474632763709778
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.8147 | LCC = 0.2602 | SRCC = 0.2546 ]
|
| 71 |
+
INFO:root:Average loss=0.7406386037667593
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.8054 | LCC = 0.2618 | SRCC = 0.2536 ]
|
| 74 |
+
INFO:root:Average loss=0.7379722109971902
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.8150 | LCC = 0.2635 | SRCC = 0.2614 ]
|
| 77 |
+
INFO:root:Average loss=0.7341487545233506
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.8073 | LCC = 0.2630 | SRCC = 0.2580 ]
|
| 80 |
+
INFO:root:Average loss=0.7340960743335577
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.7963 | LCC = 0.2616 | SRCC = 0.2544 ]
|
| 83 |
+
INFO:root:Average loss=0.7333975468690579
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.8081 | LCC = 0.2641 | SRCC = 0.2592 ]
|
| 86 |
+
INFO:root:Average loss=0.7372842583900843
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.8026 | LCC = 0.2604 | SRCC = 0.2548 ]
|
| 89 |
+
INFO:root:Average loss=0.7322733646784073
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.8087 | LCC = 0.2649 | SRCC = 0.2605 ]
|
| 92 |
+
INFO:root:Average loss=0.7382883513585116
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.7805 | LCC = 0.2660 | SRCC = 0.2607 ]
|
| 95 |
+
INFO:root:Average loss=0.7296412785847982
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.7854 | LCC = 0.2664 | SRCC = 0.2601 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.7988 | LCC = 0.2486 | SRCC = 0.2584 ]
|
results_bvcc/w2v2_large_xlsr53/layer4/run0/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer4/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0d17ddd1fc840a4c8efb2d94b5c79a8182ea133d0da68d1c78d08949e45d54f9
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer4/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c77faa9c79a76bc831692b5f9b595ac4ccb9406a66d60eb59c0a3cdc68cdec7d
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer4/run0/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.547156467460669
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3832 | LCC = 0.7870 | SRCC = 0.7937 ]
|
| 8 |
+
INFO:root:Average loss=0.29149841994811326
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3145 | LCC = 0.8253 | SRCC = 0.8260 ]
|
| 11 |
+
INFO:root:Average loss=0.24849581298155662
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2810 | LCC = 0.8434 | SRCC = 0.8447 ]
|
| 14 |
+
INFO:root:Average loss=0.22738109395290032
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2621 | LCC = 0.8573 | SRCC = 0.8536 ]
|
| 17 |
+
INFO:root:Average loss=0.21490304286663348
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2838 | LCC = 0.8643 | SRCC = 0.8606 ]
|
| 20 |
+
INFO:root:Average loss=0.20760062088569006
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2279 | LCC = 0.8673 | SRCC = 0.8642 ]
|
| 23 |
+
INFO:root:Average loss=0.20366449520373955
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2377 | LCC = 0.8735 | SRCC = 0.8702 ]
|
| 26 |
+
INFO:root:Average loss=0.17576329687084907
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2163 | LCC = 0.8735 | SRCC = 0.8715 ]
|
| 29 |
+
INFO:root:Average loss=0.17690044412246117
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2238 | LCC = 0.8779 | SRCC = 0.8757 ]
|
| 32 |
+
INFO:root:Average loss=0.1625153881808122
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2058 | LCC = 0.8757 | SRCC = 0.8747 ]
|
| 35 |
+
INFO:root:Average loss=0.16646776797297674
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2525 | LCC = 0.8692 | SRCC = 0.8712 ]
|
| 38 |
+
INFO:root:Average loss=0.16911816062071386
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2083 | LCC = 0.8794 | SRCC = 0.8774 ]
|
| 41 |
+
INFO:root:Average loss=0.14906550905643365
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2191 | LCC = 0.8795 | SRCC = 0.8771 ]
|
| 44 |
+
INFO:root:Average loss=0.14371993793891027
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2117 | LCC = 0.8791 | SRCC = 0.8769 ]
|
| 47 |
+
INFO:root:Average loss=0.1384089978841635
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2030 | LCC = 0.8787 | SRCC = 0.8776 ]
|
| 50 |
+
INFO:root:Average loss=0.12990750276889557
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2243 | LCC = 0.8789 | SRCC = 0.8775 ]
|
| 53 |
+
INFO:root:Average loss=0.13416247537884957
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1994 | LCC = 0.8776 | SRCC = 0.8764 ]
|
| 56 |
+
INFO:root:Average loss=0.1280193323126206
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2107 | LCC = 0.8757 | SRCC = 0.8748 ]
|
| 59 |
+
INFO:root:Average loss=0.11507375280444439
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2014 | LCC = 0.8787 | SRCC = 0.8778 ]
|
| 62 |
+
INFO:root:Average loss=0.11986773652143967
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2211 | LCC = 0.8760 | SRCC = 0.8761 ]
|
| 65 |
+
INFO:root:Average loss=0.10640933001652741
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2026 | LCC = 0.8763 | SRCC = 0.8763 ]
|
| 68 |
+
INFO:root:Average loss=0.10712833387347367
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1972 | LCC = 0.8812 | SRCC = 0.8802 ]
|
| 71 |
+
INFO:root:Average loss=0.10519853802636647
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2054 | LCC = 0.8747 | SRCC = 0.8757 ]
|
| 74 |
+
INFO:root:Average loss=0.09490100838817082
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2048 | LCC = 0.8798 | SRCC = 0.8781 ]
|
| 77 |
+
INFO:root:Average loss=0.09653763964963265
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2104 | LCC = 0.8747 | SRCC = 0.8743 ]
|
| 80 |
+
INFO:root:Average loss=0.09158640760832872
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2154 | LCC = 0.8728 | SRCC = 0.8724 ]
|
| 83 |
+
INFO:root:Average loss=0.08465714289400822
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2098 | LCC = 0.8769 | SRCC = 0.8759 ]
|
| 86 |
+
INFO:root:Average loss=0.0891813850746705
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2106 | LCC = 0.8787 | SRCC = 0.8780 ]
|
| 89 |
+
INFO:root:Average loss=0.09102073861047244
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2081 | LCC = 0.8761 | SRCC = 0.8754 ]
|
| 92 |
+
INFO:root:Average loss=0.08620971317092578
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2109 | LCC = 0.8755 | SRCC = 0.8746 ]
|
| 95 |
+
INFO:root:Average loss=0.07545238857467969
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2072 | LCC = 0.8736 | SRCC = 0.8727 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2096 | LCC = 0.8749 | SRCC = 0.8742 ]
|
results_bvcc/w2v2_large_xlsr53/layer4/run1/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer4/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf2cd0da48458a8dad5a0439c598dc91722a10bd72eb1293d122b1b3df500b43
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer4/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ce6fd6a252785a6f331f9946f5b53c13f73e0ee5bc262d421079252183450b33
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer4/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5605818294943907
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3382 | LCC = 0.7913 | SRCC = 0.7967 ]
|
| 8 |
+
INFO:root:Average loss=0.30025573762563557
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3533 | LCC = 0.8229 | SRCC = 0.8292 ]
|
| 11 |
+
INFO:root:Average loss=0.2813934796513655
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2893 | LCC = 0.8451 | SRCC = 0.8444 ]
|
| 14 |
+
INFO:root:Average loss=0.2493511744034596
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2820 | LCC = 0.8520 | SRCC = 0.8513 ]
|
| 17 |
+
INFO:root:Average loss=0.22732358005566475
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2426 | LCC = 0.8572 | SRCC = 0.8562 ]
|
| 20 |
+
INFO:root:Average loss=0.2069613333695974
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2915 | LCC = 0.8623 | SRCC = 0.8646 ]
|
| 23 |
+
INFO:root:Average loss=0.19090187750183618
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2433 | LCC = 0.8679 | SRCC = 0.8667 ]
|
| 26 |
+
INFO:root:Average loss=0.1918311948195482
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2164 | LCC = 0.8719 | SRCC = 0.8712 ]
|
| 29 |
+
INFO:root:Average loss=0.17321445105167535
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2449 | LCC = 0.8719 | SRCC = 0.8702 ]
|
| 32 |
+
INFO:root:Average loss=0.16294093773915216
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2188 | LCC = 0.8742 | SRCC = 0.8726 ]
|
| 35 |
+
INFO:root:Average loss=0.16203909204938474
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2208 | LCC = 0.8726 | SRCC = 0.8716 ]
|
| 38 |
+
INFO:root:Average loss=0.15740516571662364
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2292 | LCC = 0.8721 | SRCC = 0.8704 ]
|
| 41 |
+
INFO:root:Average loss=0.15466257461752647
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2270 | LCC = 0.8726 | SRCC = 0.8720 ]
|
| 44 |
+
INFO:root:Average loss=0.14520136543955559
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2052 | LCC = 0.8769 | SRCC = 0.8781 ]
|
| 47 |
+
INFO:root:Average loss=0.1345938279842719
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2260 | LCC = 0.8744 | SRCC = 0.8731 ]
|
| 50 |
+
INFO:root:Average loss=0.13297921266311255
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2284 | LCC = 0.8768 | SRCC = 0.8740 ]
|
| 53 |
+
INFO:root:Average loss=0.1225662044225595
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2110 | LCC = 0.8779 | SRCC = 0.8772 ]
|
| 56 |
+
INFO:root:Average loss=0.11552520105854058
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2273 | LCC = 0.8745 | SRCC = 0.8736 ]
|
| 59 |
+
INFO:root:Average loss=0.12532876174037272
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2443 | LCC = 0.8745 | SRCC = 0.8753 ]
|
| 62 |
+
INFO:root:Average loss=0.11093265993090776
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2307 | LCC = 0.8756 | SRCC = 0.8744 ]
|
| 65 |
+
INFO:root:Average loss=0.10863688277701537
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2385 | LCC = 0.8771 | SRCC = 0.8765 ]
|
| 68 |
+
INFO:root:Average loss=0.10018666957815488
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2663 | LCC = 0.8731 | SRCC = 0.8738 ]
|
| 71 |
+
INFO:root:Average loss=0.10150594340685086
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2035 | LCC = 0.8754 | SRCC = 0.8743 ]
|
| 74 |
+
INFO:root:Average loss=0.09620927025874455
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2666 | LCC = 0.8717 | SRCC = 0.8743 ]
|
| 77 |
+
INFO:root:Average loss=0.08765348242834592
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2430 | LCC = 0.8761 | SRCC = 0.8768 ]
|
| 80 |
+
INFO:root:Average loss=0.09197935820198976
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2052 | LCC = 0.8761 | SRCC = 0.8753 ]
|
| 83 |
+
INFO:root:Average loss=0.08101947677250092
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2142 | LCC = 0.8765 | SRCC = 0.8780 ]
|
| 86 |
+
INFO:root:Average loss=0.07929057675676468
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2166 | LCC = 0.8781 | SRCC = 0.8765 ]
|
| 89 |
+
INFO:root:Average loss=0.08396360980203518
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2919 | LCC = 0.8682 | SRCC = 0.8720 ]
|
| 92 |
+
INFO:root:Average loss=0.074722120204033
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2033 | LCC = 0.8764 | SRCC = 0.8761 ]
|
| 95 |
+
INFO:root:Average loss=0.0711737318107715
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2367 | LCC = 0.8755 | SRCC = 0.8754 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2166 | LCC = 0.8761 | SRCC = 0.8773 ]
|
results_bvcc/w2v2_large_xlsr53/layer4/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer4/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4988351cc534c94f99f7559f933edb0081f47cafa055d2a3e17c09d9ed472b3a
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer4/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ababe9b3184cb9bc8887520d5ba3bf14d8176e8d09fbc642ac6cc11c9daddcd0
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer4/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.4532563615685854
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3356 | LCC = 0.7989 | SRCC = 0.8013 ]
|
| 8 |
+
INFO:root:Average loss=0.27491417641823107
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2828 | LCC = 0.8347 | SRCC = 0.8370 ]
|
| 11 |
+
INFO:root:Average loss=0.2513206410102355
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2710 | LCC = 0.8484 | SRCC = 0.8482 ]
|
| 14 |
+
INFO:root:Average loss=0.2428954717440483
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2799 | LCC = 0.8471 | SRCC = 0.8524 ]
|
| 17 |
+
INFO:root:Average loss=0.21514941904789361
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2534 | LCC = 0.8650 | SRCC = 0.8626 ]
|
| 20 |
+
INFO:root:Average loss=0.2048763797069207
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2392 | LCC = 0.8661 | SRCC = 0.8668 ]
|
| 23 |
+
INFO:root:Average loss=0.19460664336115885
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2166 | LCC = 0.8683 | SRCC = 0.8700 ]
|
| 26 |
+
INFO:root:Average loss=0.18061789364004746
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2214 | LCC = 0.8693 | SRCC = 0.8684 ]
|
| 29 |
+
INFO:root:Average loss=0.18510188811864609
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2186 | LCC = 0.8743 | SRCC = 0.8720 ]
|
| 32 |
+
INFO:root:Average loss=0.1711409091949463
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2395 | LCC = 0.8728 | SRCC = 0.8728 ]
|
| 35 |
+
INFO:root:Average loss=0.1823367731502423
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2216 | LCC = 0.8695 | SRCC = 0.8729 ]
|
| 38 |
+
INFO:root:Average loss=0.15918540228635836
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2110 | LCC = 0.8734 | SRCC = 0.8718 ]
|
| 41 |
+
INFO:root:Average loss=0.15191630302713469
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2003 | LCC = 0.8755 | SRCC = 0.8753 ]
|
| 44 |
+
INFO:root:Average loss=0.15265449738273254
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2462 | LCC = 0.8677 | SRCC = 0.8726 ]
|
| 47 |
+
INFO:root:Average loss=0.1382217956468081
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2130 | LCC = 0.8727 | SRCC = 0.8747 ]
|
| 50 |
+
INFO:root:Average loss=0.14303912518498224
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2350 | LCC = 0.8722 | SRCC = 0.8728 ]
|
| 53 |
+
INFO:root:Average loss=0.13195646449159354
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2163 | LCC = 0.8779 | SRCC = 0.8764 ]
|
| 56 |
+
INFO:root:Average loss=0.12995244027712408
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2148 | LCC = 0.8735 | SRCC = 0.8752 ]
|
| 59 |
+
INFO:root:Average loss=0.12340931288706951
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2682 | LCC = 0.8657 | SRCC = 0.8737 ]
|
| 62 |
+
INFO:root:Average loss=0.11715158571799596
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2107 | LCC = 0.8726 | SRCC = 0.8745 ]
|
| 65 |
+
INFO:root:Average loss=0.1106030118580048
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2178 | LCC = 0.8728 | SRCC = 0.8728 ]
|
| 68 |
+
INFO:root:Average loss=0.11555583459826615
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2155 | LCC = 0.8780 | SRCC = 0.8771 ]
|
| 71 |
+
INFO:root:Average loss=0.10848726771580867
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2078 | LCC = 0.8758 | SRCC = 0.8753 ]
|
| 74 |
+
INFO:root:Average loss=0.09563501962484458
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2425 | LCC = 0.8681 | SRCC = 0.8676 ]
|
| 77 |
+
INFO:root:Average loss=0.09557310367623965
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2283 | LCC = 0.8726 | SRCC = 0.8758 ]
|
| 80 |
+
INFO:root:Average loss=0.09602287803322841
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1953 | LCC = 0.8783 | SRCC = 0.8786 ]
|
| 83 |
+
INFO:root:Average loss=0.10340599371836735
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2294 | LCC = 0.8741 | SRCC = 0.8754 ]
|
| 86 |
+
INFO:root:Average loss=0.08820458687841892
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2308 | LCC = 0.8733 | SRCC = 0.8737 ]
|
| 89 |
+
INFO:root:Average loss=0.08390403567598416
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2481 | LCC = 0.8685 | SRCC = 0.8749 ]
|
| 92 |
+
INFO:root:Average loss=0.0964965601093494
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1953 | LCC = 0.8769 | SRCC = 0.8774 ]
|
| 95 |
+
INFO:root:Average loss=0.08484634895546314
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2053 | LCC = 0.8762 | SRCC = 0.8762 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2247 | LCC = 0.8717 | SRCC = 0.8723 ]
|
results_bvcc/w2v2_large_xlsr53/layer4/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer4/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:727657e5e90f3d64442de165ff4ede62d7b7457f843b30dba402b9401a31d0b3
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer4/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b1f800e6f5de54cefe7b88d3f824da19de7ff6faf367c840ec7f221ca3160784
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_xlsr53/layer4/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.4696754650809826
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3065 | LCC = 0.8023 | SRCC = 0.8093 ]
|
| 8 |
+
INFO:root:Average loss=0.2954771388799716
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2991 | LCC = 0.8411 | SRCC = 0.8407 ]
|
| 11 |
+
INFO:root:Average loss=0.28332846611738205
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2869 | LCC = 0.8463 | SRCC = 0.8464 ]
|
| 14 |
+
INFO:root:Average loss=0.23383696663838166
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2663 | LCC = 0.8586 | SRCC = 0.8569 ]
|
| 17 |
+
INFO:root:Average loss=0.21505269962243545
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2557 | LCC = 0.8593 | SRCC = 0.8594 ]
|
| 20 |
+
INFO:root:Average loss=0.21830975111478415
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2455 | LCC = 0.8675 | SRCC = 0.8651 ]
|
| 23 |
+
INFO:root:Average loss=0.1987616410240149
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2317 | LCC = 0.8693 | SRCC = 0.8681 ]
|
| 26 |
+
INFO:root:Average loss=0.19079558522655413
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2670 | LCC = 0.8720 | SRCC = 0.8704 ]
|
| 29 |
+
INFO:root:Average loss=0.18426570716576698
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2334 | LCC = 0.8724 | SRCC = 0.8717 ]
|
| 32 |
+
INFO:root:Average loss=0.17354323495274934
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2210 | LCC = 0.8753 | SRCC = 0.8748 ]
|
| 35 |
+
INFO:root:Average loss=0.18379609582897943
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2884 | LCC = 0.8663 | SRCC = 0.8725 ]
|
| 38 |
+
INFO:root:Average loss=0.1759594639715476
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2341 | LCC = 0.8718 | SRCC = 0.8731 ]
|
| 41 |
+
INFO:root:Average loss=0.16178548221404737
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2050 | LCC = 0.8786 | SRCC = 0.8777 ]
|
| 44 |
+
INFO:root:Average loss=0.14725517749022216
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2202 | LCC = 0.8749 | SRCC = 0.8736 ]
|
| 47 |
+
INFO:root:Average loss=0.1469860810499925
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2186 | LCC = 0.8779 | SRCC = 0.8773 ]
|
| 50 |
+
INFO:root:Average loss=0.1449679702711411
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1947 | LCC = 0.8786 | SRCC = 0.8785 ]
|
| 53 |
+
INFO:root:Average loss=0.13312828435729712
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2659 | LCC = 0.8658 | SRCC = 0.8687 ]
|
| 56 |
+
INFO:root:Average loss=0.1286121530410571
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2298 | LCC = 0.8746 | SRCC = 0.8741 ]
|
| 59 |
+
INFO:root:Average loss=0.12523656710982323
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2136 | LCC = 0.8763 | SRCC = 0.8749 ]
|
| 62 |
+
INFO:root:Average loss=0.12012366854991668
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2106 | LCC = 0.8773 | SRCC = 0.8772 ]
|
| 65 |
+
INFO:root:Average loss=0.11537392714466804
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2178 | LCC = 0.8739 | SRCC = 0.8743 ]
|
| 68 |
+
INFO:root:Average loss=0.10499710245774342
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2007 | LCC = 0.8805 | SRCC = 0.8794 ]
|
| 71 |
+
INFO:root:Average loss=0.10581345703357305
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2155 | LCC = 0.8743 | SRCC = 0.8756 ]
|
| 74 |
+
INFO:root:Average loss=0.10889474092385708
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2006 | LCC = 0.8767 | SRCC = 0.8754 ]
|
| 77 |
+
INFO:root:Average loss=0.1006790762528395
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2121 | LCC = 0.8718 | SRCC = 0.8693 ]
|
| 80 |
+
INFO:root:Average loss=0.09053596796897742
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2317 | LCC = 0.8743 | SRCC = 0.8740 ]
|
| 83 |
+
INFO:root:Average loss=0.09128491265269426
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2291 | LCC = 0.8756 | SRCC = 0.8734 ]
|
| 86 |
+
INFO:root:Average loss=0.08415590680371492
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2174 | LCC = 0.8759 | SRCC = 0.8760 ]
|
| 89 |
+
INFO:root:Average loss=0.08523786067962646
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2122 | LCC = 0.8769 | SRCC = 0.8763 ]
|
| 92 |
+
INFO:root:Average loss=0.08090076237343825
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2105 | LCC = 0.8771 | SRCC = 0.8764 ]
|
| 95 |
+
INFO:root:Average loss=0.0761953240308242
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2198 | LCC = 0.8731 | SRCC = 0.8756 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2211 | LCC = 0.8703 | SRCC = 0.8700 ]
|
results_bvcc/w2v2_large_xlsr53/layer4/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_xlsr53/layer4/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f32c3616db4d0fbebd275bee428cbbddb517f648dd559fd48498d502ae6040be
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_xlsr53/layer4/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:904fa759cb4ec83f126ca39d143affdeea20223e7a8c03010188a94ce91d7de4
|
| 3 |
+
size 3405062
|