Upload SSL Layer MOS checkpoints (part 4)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run4/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run4/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run0/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run0/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run0/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run0/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run1/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run1/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run1/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run1/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run2/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run4/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run4/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run0/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run0/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run0/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run0/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run1/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run1/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run1/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run1/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run2/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run4/model_best.pt +3 -0
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4e4b0ad4d78054b6c88cc06c2a2d60197c98e996998b4b3e23e7ed4e6e9135f
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:134a826c34e22c6a8fca0a5bf8bd2a487d03cc5cdfd3d50e848e6d54e7424eb2
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/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.6006987996590443
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4035 | LCC = 0.7393 | SRCC = 0.7421 ]
|
| 8 |
+
INFO:root:Average loss=0.3137065510337169
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3200 | LCC = 0.8046 | SRCC = 0.8113 ]
|
| 11 |
+
INFO:root:Average loss=0.2671936918527652
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2864 | LCC = 0.8191 | SRCC = 0.8263 ]
|
| 14 |
+
INFO:root:Average loss=0.23879283322737768
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2656 | LCC = 0.8378 | SRCC = 0.8368 ]
|
| 17 |
+
INFO:root:Average loss=0.2259096677104632
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2962 | LCC = 0.8344 | SRCC = 0.8401 ]
|
| 20 |
+
INFO:root:Average loss=0.21637474974760643
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2990 | LCC = 0.8439 | SRCC = 0.8460 ]
|
| 23 |
+
INFO:root:Average loss=0.19388003131517997
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2436 | LCC = 0.8568 | SRCC = 0.8550 ]
|
| 26 |
+
INFO:root:Average loss=0.19029025418254045
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2536 | LCC = 0.8575 | SRCC = 0.8558 ]
|
| 29 |
+
INFO:root:Average loss=0.18381341470357698
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2399 | LCC = 0.8521 | SRCC = 0.8534 ]
|
| 32 |
+
INFO:root:Average loss=0.1676634604541155
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2438 | LCC = 0.8603 | SRCC = 0.8580 ]
|
| 35 |
+
INFO:root:Average loss=0.15254133318861327
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2648 | LCC = 0.8552 | SRCC = 0.8573 ]
|
| 38 |
+
INFO:root:Average loss=0.16495537557280981
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2461 | LCC = 0.8611 | SRCC = 0.8600 ]
|
| 41 |
+
INFO:root:Average loss=0.1467983495348539
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2518 | LCC = 0.8587 | SRCC = 0.8578 ]
|
| 44 |
+
INFO:root:Average loss=0.14043838798235625
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2561 | LCC = 0.8601 | SRCC = 0.8600 ]
|
| 47 |
+
INFO:root:Average loss=0.1383707364782309
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2628 | LCC = 0.8569 | SRCC = 0.8559 ]
|
| 50 |
+
INFO:root:Average loss=0.1251800062182622
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2177 | LCC = 0.8619 | SRCC = 0.8603 ]
|
| 53 |
+
INFO:root:Average loss=0.11802030803683476
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3029 | LCC = 0.8581 | SRCC = 0.8580 ]
|
| 56 |
+
INFO:root:Average loss=0.1248860084093534
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2571 | LCC = 0.8634 | SRCC = 0.8607 ]
|
| 59 |
+
INFO:root:Average loss=0.11575037221878003
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2250 | LCC = 0.8592 | SRCC = 0.8571 ]
|
| 62 |
+
INFO:root:Average loss=0.11726610811474995
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2842 | LCC = 0.8518 | SRCC = 0.8499 ]
|
| 65 |
+
INFO:root:Average loss=0.10158162525831124
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2816 | LCC = 0.8557 | SRCC = 0.8544 ]
|
| 68 |
+
INFO:root:Average loss=0.10218472902973492
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2462 | LCC = 0.8570 | SRCC = 0.8523 ]
|
| 71 |
+
INFO:root:Average loss=0.09679935595546013
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2275 | LCC = 0.8571 | SRCC = 0.8536 ]
|
| 74 |
+
INFO:root:Average loss=0.08840126854677995
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2408 | LCC = 0.8610 | SRCC = 0.8589 ]
|
| 77 |
+
INFO:root:Average loss=0.08986751415217534
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2866 | LCC = 0.8540 | SRCC = 0.8524 ]
|
| 80 |
+
INFO:root:Average loss=0.09172978113668087
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2293 | LCC = 0.8573 | SRCC = 0.8543 ]
|
| 83 |
+
INFO:root:Average loss=0.08408539570294894
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2987 | LCC = 0.8526 | SRCC = 0.8531 ]
|
| 86 |
+
INFO:root:Average loss=0.07780154694158298
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2972 | LCC = 0.8521 | SRCC = 0.8531 ]
|
| 89 |
+
INFO:root:Average loss=0.0770007719596227
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2529 | LCC = 0.8601 | SRCC = 0.8591 ]
|
| 92 |
+
INFO:root:Average loss=0.08227077680520523
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2347 | LCC = 0.8619 | SRCC = 0.8597 ]
|
| 95 |
+
INFO:root:Average loss=0.07572061382234097
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2232 | LCC = 0.8599 | SRCC = 0.8564 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2355 | LCC = 0.8661 | SRCC = 0.8644 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/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_base_bvcc_finetuned_1epoch/layer1/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c636624e53d1c0afa3183a58996ded79e592b348187fcd15deda685d927d29a0
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e51b115bb9b02bb48305274413ef23770ee01eae941ba4e752dbb16ad019d3c9
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/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.5577032099931668
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3960 | LCC = 0.7473 | SRCC = 0.7570 ]
|
| 8 |
+
INFO:root:Average loss=0.3139839286987598
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3711 | LCC = 0.7898 | SRCC = 0.8000 ]
|
| 11 |
+
INFO:root:Average loss=0.26633496028490555
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2800 | LCC = 0.8172 | SRCC = 0.8215 ]
|
| 14 |
+
INFO:root:Average loss=0.23809264447444525
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2644 | LCC = 0.8336 | SRCC = 0.8349 ]
|
| 17 |
+
INFO:root:Average loss=0.22117552180320788
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2862 | LCC = 0.8385 | SRCC = 0.8441 ]
|
| 20 |
+
INFO:root:Average loss=0.22384615643666342
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2635 | LCC = 0.8399 | SRCC = 0.8414 ]
|
| 23 |
+
INFO:root:Average loss=0.196821590169118
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2531 | LCC = 0.8515 | SRCC = 0.8517 ]
|
| 26 |
+
INFO:root:Average loss=0.1905044674491271
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2389 | LCC = 0.8539 | SRCC = 0.8538 ]
|
| 29 |
+
INFO:root:Average loss=0.1764051616191864
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2429 | LCC = 0.8567 | SRCC = 0.8578 ]
|
| 32 |
+
INFO:root:Average loss=0.1762571218304145
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2372 | LCC = 0.8569 | SRCC = 0.8569 ]
|
| 35 |
+
INFO:root:Average loss=0.1647373527670518
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2341 | LCC = 0.8606 | SRCC = 0.8609 ]
|
| 38 |
+
INFO:root:Average loss=0.14897103540790388
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2563 | LCC = 0.8568 | SRCC = 0.8567 ]
|
| 41 |
+
INFO:root:Average loss=0.14665505738976675
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2752 | LCC = 0.8621 | SRCC = 0.8635 ]
|
| 44 |
+
INFO:root:Average loss=0.14233640380776846
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2302 | LCC = 0.8609 | SRCC = 0.8600 ]
|
| 47 |
+
INFO:root:Average loss=0.13174179072181383
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2480 | LCC = 0.8584 | SRCC = 0.8587 ]
|
| 50 |
+
INFO:root:Average loss=0.12609843747356
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2196 | LCC = 0.8611 | SRCC = 0.8603 ]
|
| 53 |
+
INFO:root:Average loss=0.12385736807034566
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2273 | LCC = 0.8609 | SRCC = 0.8597 ]
|
| 56 |
+
INFO:root:Average loss=0.12338605494453357
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2211 | LCC = 0.8622 | SRCC = 0.8624 ]
|
| 59 |
+
INFO:root:Average loss=0.11610402061771123
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2434 | LCC = 0.8593 | SRCC = 0.8601 ]
|
| 62 |
+
INFO:root:Average loss=0.10993184798803085
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2175 | LCC = 0.8614 | SRCC = 0.8616 ]
|
| 65 |
+
INFO:root:Average loss=0.10942104134040001
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2827 | LCC = 0.8608 | SRCC = 0.8638 ]
|
| 68 |
+
INFO:root:Average loss=0.10266295400185463
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2215 | LCC = 0.8581 | SRCC = 0.8576 ]
|
| 71 |
+
INFO:root:Average loss=0.09171454837689033
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2152 | LCC = 0.8604 | SRCC = 0.8597 ]
|
| 74 |
+
INFO:root:Average loss=0.09755968875609912
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2246 | LCC = 0.8598 | SRCC = 0.8579 ]
|
| 77 |
+
INFO:root:Average loss=0.08820604752653684
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2275 | LCC = 0.8575 | SRCC = 0.8584 ]
|
| 80 |
+
INFO:root:Average loss=0.09225147413328671
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2402 | LCC = 0.8590 | SRCC = 0.8582 ]
|
| 83 |
+
INFO:root:Average loss=0.08601127798931721
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2248 | LCC = 0.8610 | SRCC = 0.8600 ]
|
| 86 |
+
INFO:root:Average loss=0.08617788347869347
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2310 | LCC = 0.8594 | SRCC = 0.8588 ]
|
| 89 |
+
INFO:root:Average loss=0.08935599688154
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2203 | LCC = 0.8642 | SRCC = 0.8635 ]
|
| 92 |
+
INFO:root:Average loss=0.08155740830951776
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2393 | LCC = 0.8626 | SRCC = 0.8622 ]
|
| 95 |
+
INFO:root:Average loss=0.0741566596313929
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2308 | LCC = 0.8627 | SRCC = 0.8617 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2517 | LCC = 0.8589 | SRCC = 0.8608 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/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_base_bvcc_finetuned_1epoch/layer1/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9998b5a682cc919f1674f782e52a2a8c8441c3bee308a80ce7615d5310d59aab
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:29e10179f34e76d38dccebd4a19ea7ec650289e330f4368acd0419332be11647
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer1/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.5525378086245977
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3801 | LCC = 0.7480 | SRCC = 0.7594 ]
|
| 8 |
+
INFO:root:Average loss=0.3283146917819977
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3379 | LCC = 0.7939 | SRCC = 0.7984 ]
|
| 11 |
+
INFO:root:Average loss=0.2786462990901409
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3004 | LCC = 0.8176 | SRCC = 0.8213 ]
|
| 14 |
+
INFO:root:Average loss=0.24937959588491
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2951 | LCC = 0.8254 | SRCC = 0.8301 ]
|
| 17 |
+
INFO:root:Average loss=0.22587873032077765
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2642 | LCC = 0.8376 | SRCC = 0.8403 ]
|
| 20 |
+
INFO:root:Average loss=0.22421716019893304
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2549 | LCC = 0.8432 | SRCC = 0.8443 ]
|
| 23 |
+
INFO:root:Average loss=0.1979722100763749
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2470 | LCC = 0.8482 | SRCC = 0.8487 ]
|
| 26 |
+
INFO:root:Average loss=0.1896534298474972
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2490 | LCC = 0.8540 | SRCC = 0.8532 ]
|
| 29 |
+
INFO:root:Average loss=0.18755720479366106
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2494 | LCC = 0.8522 | SRCC = 0.8527 ]
|
| 32 |
+
INFO:root:Average loss=0.1769320450914212
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2816 | LCC = 0.8599 | SRCC = 0.8602 ]
|
| 35 |
+
INFO:root:Average loss=0.1606948836109577
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2416 | LCC = 0.8544 | SRCC = 0.8563 ]
|
| 38 |
+
INFO:root:Average loss=0.15922243253160745
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2432 | LCC = 0.8520 | SRCC = 0.8570 ]
|
| 41 |
+
INFO:root:Average loss=0.15271418493909714
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2739 | LCC = 0.8622 | SRCC = 0.8626 ]
|
| 44 |
+
INFO:root:Average loss=0.13986208490454233
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2309 | LCC = 0.8565 | SRCC = 0.8570 ]
|
| 47 |
+
INFO:root:Average loss=0.13474046658629027
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2371 | LCC = 0.8566 | SRCC = 0.8574 ]
|
| 50 |
+
INFO:root:Average loss=0.13050641730809823
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2291 | LCC = 0.8577 | SRCC = 0.8600 ]
|
| 53 |
+
INFO:root:Average loss=0.13568030297756195
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2652 | LCC = 0.8479 | SRCC = 0.8506 ]
|
| 56 |
+
INFO:root:Average loss=0.1308771126354352
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2555 | LCC = 0.8565 | SRCC = 0.8586 ]
|
| 59 |
+
INFO:root:Average loss=0.11732197944552471
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2207 | LCC = 0.8625 | SRCC = 0.8624 ]
|
| 62 |
+
INFO:root:Average loss=0.11015717618358441
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2351 | LCC = 0.8544 | SRCC = 0.8545 ]
|
| 65 |
+
INFO:root:Average loss=0.11742200769293003
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2320 | LCC = 0.8560 | SRCC = 0.8550 ]
|
| 68 |
+
INFO:root:Average loss=0.10351483776974373
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2205 | LCC = 0.8598 | SRCC = 0.8599 ]
|
| 71 |
+
INFO:root:Average loss=0.09843618785723662
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2249 | LCC = 0.8631 | SRCC = 0.8629 ]
|
| 74 |
+
INFO:root:Average loss=0.09560239171752563
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2195 | LCC = 0.8619 | SRCC = 0.8623 ]
|
| 77 |
+
INFO:root:Average loss=0.09047879804021274
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2188 | LCC = 0.8632 | SRCC = 0.8631 ]
|
| 80 |
+
INFO:root:Average loss=0.08644061974990062
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2262 | LCC = 0.8628 | SRCC = 0.8638 ]
|
| 83 |
+
INFO:root:Average loss=0.08795356654968017
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2242 | LCC = 0.8632 | SRCC = 0.8626 ]
|
| 86 |
+
INFO:root:Average loss=0.08254784541443372
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2226 | LCC = 0.8636 | SRCC = 0.8642 ]
|
| 89 |
+
INFO:root:Average loss=0.07980456217550315
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2352 | LCC = 0.8621 | SRCC = 0.8624 ]
|
| 92 |
+
INFO:root:Average loss=0.08143908043320362
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2126 | LCC = 0.8656 | SRCC = 0.8648 ]
|
| 95 |
+
INFO:root:Average loss=0.07162522601011471
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2210 | LCC = 0.8644 | SRCC = 0.8631 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2354 | LCC = 0.8562 | SRCC = 0.8566 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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_base_bvcc_finetuned_1epoch/layer10/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c01b85956dd2c265e049fd027abc53232b8fcac639ae292c57d90a2c30d3e84c
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7191ad77c6b4b6e91368e9805e5e5a4605c975fdbb33f86101ff06ad90581abe
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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.5794957517049252
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4068 | LCC = 0.7381 | SRCC = 0.7421 ]
|
| 8 |
+
INFO:root:Average loss=0.4189493446013866
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4210 | LCC = 0.7564 | SRCC = 0.7648 ]
|
| 11 |
+
INFO:root:Average loss=0.36668664350723607
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3434 | LCC = 0.7733 | SRCC = 0.7771 ]
|
| 14 |
+
INFO:root:Average loss=0.3401531764330008
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.4237 | LCC = 0.7830 | SRCC = 0.7853 ]
|
| 17 |
+
INFO:root:Average loss=0.37762189331727153
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3745 | LCC = 0.7919 | SRCC = 0.7934 ]
|
| 20 |
+
INFO:root:Average loss=0.34444289979262227
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3363 | LCC = 0.7900 | SRCC = 0.7927 ]
|
| 23 |
+
INFO:root:Average loss=0.3259281334586633
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3513 | LCC = 0.7941 | SRCC = 0.7969 ]
|
| 26 |
+
INFO:root:Average loss=0.3640765360532663
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3682 | LCC = 0.7982 | SRCC = 0.8001 ]
|
| 29 |
+
INFO:root:Average loss=0.3193130762531207
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3944 | LCC = 0.8038 | SRCC = 0.8028 ]
|
| 32 |
+
INFO:root:Average loss=0.31694377423861086
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3548 | LCC = 0.8045 | SRCC = 0.8080 ]
|
| 35 |
+
INFO:root:Average loss=0.2900777955849965
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3224 | LCC = 0.8106 | SRCC = 0.8107 ]
|
| 38 |
+
INFO:root:Average loss=0.30088877563293165
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3669 | LCC = 0.8102 | SRCC = 0.8106 ]
|
| 41 |
+
INFO:root:Average loss=0.31550150078076583
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3481 | LCC = 0.8067 | SRCC = 0.8133 ]
|
| 44 |
+
INFO:root:Average loss=0.32858273272330946
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3460 | LCC = 0.8098 | SRCC = 0.8102 ]
|
| 47 |
+
INFO:root:Average loss=0.29860056879428715
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3120 | LCC = 0.8110 | SRCC = 0.8130 ]
|
| 50 |
+
INFO:root:Average loss=0.28414352238178253
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3954 | LCC = 0.8114 | SRCC = 0.8155 ]
|
| 53 |
+
INFO:root:Average loss=0.3009450370684648
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3076 | LCC = 0.8176 | SRCC = 0.8183 ]
|
| 56 |
+
INFO:root:Average loss=0.28250264758482957
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3151 | LCC = 0.8181 | SRCC = 0.8192 ]
|
| 59 |
+
INFO:root:Average loss=0.28335456511913204
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2960 | LCC = 0.8216 | SRCC = 0.8225 ]
|
| 62 |
+
INFO:root:Average loss=0.27502379814783734
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3184 | LCC = 0.8235 | SRCC = 0.8223 ]
|
| 65 |
+
INFO:root:Average loss=0.28518662066795886
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3113 | LCC = 0.8196 | SRCC = 0.8211 ]
|
| 68 |
+
INFO:root:Average loss=0.2678028616385582
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3142 | LCC = 0.8194 | SRCC = 0.8222 ]
|
| 71 |
+
INFO:root:Average loss=0.2691647553673157
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3001 | LCC = 0.8212 | SRCC = 0.8246 ]
|
| 74 |
+
INFO:root:Average loss=0.26632089358873856
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3196 | LCC = 0.8226 | SRCC = 0.8241 ]
|
| 77 |
+
INFO:root:Average loss=0.25906851028020567
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2901 | LCC = 0.8264 | SRCC = 0.8262 ]
|
| 80 |
+
INFO:root:Average loss=0.27062089913166487
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3355 | LCC = 0.8223 | SRCC = 0.8220 ]
|
| 83 |
+
INFO:root:Average loss=0.2600910098124773
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3299 | LCC = 0.8237 | SRCC = 0.8261 ]
|
| 86 |
+
INFO:root:Average loss=0.2629608867260126
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3656 | LCC = 0.8260 | SRCC = 0.8273 ]
|
| 89 |
+
INFO:root:Average loss=0.26037159600319004
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2837 | LCC = 0.8263 | SRCC = 0.8271 ]
|
| 92 |
+
INFO:root:Average loss=0.2623340986095942
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3047 | LCC = 0.8281 | SRCC = 0.8288 ]
|
| 95 |
+
INFO:root:Average loss=0.2514979585241049
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2898 | LCC = 0.8320 | SRCC = 0.8312 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2730 | LCC = 0.8443 | SRCC = 0.8449 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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_base_bvcc_finetuned_1epoch/layer10/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0b794925a03f4cc7a3177521458677cee751004db5a8778837ebcad1347ea42
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:308cf8b65f03eac70da9b4a0502d70ca898dd9e9afe1f2f0fd14b80c8d558a4a
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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.5769437398665991
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5157 | LCC = 0.7236 | SRCC = 0.7292 ]
|
| 8 |
+
INFO:root:Average loss=0.39619894574085873
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.6369 | LCC = 0.7354 | SRCC = 0.7573 ]
|
| 11 |
+
INFO:root:Average loss=0.4234802210942293
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.4320 | LCC = 0.7649 | SRCC = 0.7706 ]
|
| 14 |
+
INFO:root:Average loss=0.34623727737328946
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.4378 | LCC = 0.7664 | SRCC = 0.7802 ]
|
| 17 |
+
INFO:root:Average loss=0.35806382734041947
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3903 | LCC = 0.7893 | SRCC = 0.7910 ]
|
| 20 |
+
INFO:root:Average loss=0.34265081469829267
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3588 | LCC = 0.7943 | SRCC = 0.7971 ]
|
| 23 |
+
INFO:root:Average loss=0.3497260153675691
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3834 | LCC = 0.7989 | SRCC = 0.7980 ]
|
| 26 |
+
INFO:root:Average loss=0.3079837965659606
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3546 | LCC = 0.7959 | SRCC = 0.8022 ]
|
| 29 |
+
INFO:root:Average loss=0.34615337886871433
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3828 | LCC = 0.8016 | SRCC = 0.8015 ]
|
| 32 |
+
INFO:root:Average loss=0.32590545389132625
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3786 | LCC = 0.8042 | SRCC = 0.8030 ]
|
| 35 |
+
INFO:root:Average loss=0.31244672987705624
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3602 | LCC = 0.8035 | SRCC = 0.8060 ]
|
| 38 |
+
INFO:root:Average loss=0.30032417598443156
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3104 | LCC = 0.8084 | SRCC = 0.8090 ]
|
| 41 |
+
INFO:root:Average loss=0.2947550729299203
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3908 | LCC = 0.8032 | SRCC = 0.8076 ]
|
| 44 |
+
INFO:root:Average loss=0.3025146457247245
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3462 | LCC = 0.8093 | SRCC = 0.8115 ]
|
| 47 |
+
INFO:root:Average loss=0.3008393139029161
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3523 | LCC = 0.8120 | SRCC = 0.8133 ]
|
| 50 |
+
INFO:root:Average loss=0.28342798753426623
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2944 | LCC = 0.8133 | SRCC = 0.8164 ]
|
| 53 |
+
INFO:root:Average loss=0.2859558656047552
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3266 | LCC = 0.8179 | SRCC = 0.8187 ]
|
| 56 |
+
INFO:root:Average loss=0.2910258477697006
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3861 | LCC = 0.8145 | SRCC = 0.8193 ]
|
| 59 |
+
INFO:root:Average loss=0.2872061393199823
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3239 | LCC = 0.8148 | SRCC = 0.8149 ]
|
| 62 |
+
INFO:root:Average loss=0.27876897098926395
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2842 | LCC = 0.8175 | SRCC = 0.8201 ]
|
| 65 |
+
INFO:root:Average loss=0.3017055675005301
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3043 | LCC = 0.8213 | SRCC = 0.8217 ]
|
| 68 |
+
INFO:root:Average loss=0.2770625746402985
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3750 | LCC = 0.8132 | SRCC = 0.8160 ]
|
| 71 |
+
INFO:root:Average loss=0.28191186850651717
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3439 | LCC = 0.8216 | SRCC = 0.8208 ]
|
| 74 |
+
INFO:root:Average loss=0.2829067158775452
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3105 | LCC = 0.8190 | SRCC = 0.8208 ]
|
| 77 |
+
INFO:root:Average loss=0.257286178568999
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3301 | LCC = 0.8209 | SRCC = 0.8244 ]
|
| 80 |
+
INFO:root:Average loss=0.2666809998261623
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3797 | LCC = 0.8139 | SRCC = 0.8200 ]
|
| 83 |
+
INFO:root:Average loss=0.26847826402920943
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3254 | LCC = 0.8207 | SRCC = 0.8233 ]
|
| 86 |
+
INFO:root:Average loss=0.26848746855289507
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3341 | LCC = 0.8252 | SRCC = 0.8263 ]
|
| 89 |
+
INFO:root:Average loss=0.25932069772329086
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.4292 | LCC = 0.8149 | SRCC = 0.8217 ]
|
| 92 |
+
INFO:root:Average loss=0.25409679133922625
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3159 | LCC = 0.8272 | SRCC = 0.8283 ]
|
| 95 |
+
INFO:root:Average loss=0.278778137304844
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3522 | LCC = 0.8238 | SRCC = 0.8233 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2762 | LCC = 0.8442 | SRCC = 0.8439 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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_base_bvcc_finetuned_1epoch/layer10/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:14ea434751c8c3bfa0f3e31df79a3bce90d9313a5b8fb091f8d53eb36667666e
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dafcfe76d0c533f3fc655b9b0c9189a9b110efdd7afe0b1d5388544304b6aa05
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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.5942672498715229
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4659 | LCC = 0.7239 | SRCC = 0.7251 ]
|
| 8 |
+
INFO:root:Average loss=0.382832509966997
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4325 | LCC = 0.7512 | SRCC = 0.7651 ]
|
| 11 |
+
INFO:root:Average loss=0.3669925102820763
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3678 | LCC = 0.7743 | SRCC = 0.7784 ]
|
| 14 |
+
INFO:root:Average loss=0.3417179158482796
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.4020 | LCC = 0.7845 | SRCC = 0.7916 ]
|
| 17 |
+
INFO:root:Average loss=0.34532631360567534
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3430 | LCC = 0.7916 | SRCC = 0.7938 ]
|
| 20 |
+
INFO:root:Average loss=0.33108560951092303
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3680 | LCC = 0.8033 | SRCC = 0.8022 ]
|
| 23 |
+
INFO:root:Average loss=0.3267887274806316
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3214 | LCC = 0.8031 | SRCC = 0.8039 ]
|
| 26 |
+
INFO:root:Average loss=0.3265558108687401
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3813 | LCC = 0.7969 | SRCC = 0.8028 ]
|
| 29 |
+
INFO:root:Average loss=0.32854782083095646
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3276 | LCC = 0.8058 | SRCC = 0.8069 ]
|
| 32 |
+
INFO:root:Average loss=0.3254155149826637
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3376 | LCC = 0.8090 | SRCC = 0.8091 ]
|
| 35 |
+
INFO:root:Average loss=0.35408404450386
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.4920 | LCC = 0.8031 | SRCC = 0.8057 ]
|
| 38 |
+
INFO:root:Average loss=0.312020586278194
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3196 | LCC = 0.8076 | SRCC = 0.8113 ]
|
| 41 |
+
INFO:root:Average loss=0.3042628608452968
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3685 | LCC = 0.8150 | SRCC = 0.8135 ]
|
| 44 |
+
INFO:root:Average loss=0.2875899960979437
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3601 | LCC = 0.8030 | SRCC = 0.8116 ]
|
| 47 |
+
INFO:root:Average loss=0.33370670542503017
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3509 | LCC = 0.8135 | SRCC = 0.8145 ]
|
| 50 |
+
INFO:root:Average loss=0.2957675413061411
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.4166 | LCC = 0.8062 | SRCC = 0.8131 ]
|
| 53 |
+
INFO:root:Average loss=0.29341954948046267
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3046 | LCC = 0.8207 | SRCC = 0.8188 ]
|
| 56 |
+
INFO:root:Average loss=0.28014899484622174
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3088 | LCC = 0.8125 | SRCC = 0.8148 ]
|
| 59 |
+
INFO:root:Average loss=0.27201277628923076
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3136 | LCC = 0.8228 | SRCC = 0.8208 ]
|
| 62 |
+
INFO:root:Average loss=0.2936048278441796
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3057 | LCC = 0.8176 | SRCC = 0.8207 ]
|
| 65 |
+
INFO:root:Average loss=0.2729914942995096
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2892 | LCC = 0.8234 | SRCC = 0.8236 ]
|
| 68 |
+
INFO:root:Average loss=0.27233778494290817
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3265 | LCC = 0.8279 | SRCC = 0.8260 ]
|
| 71 |
+
INFO:root:Average loss=0.2711257313688596
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3274 | LCC = 0.8245 | SRCC = 0.8237 ]
|
| 74 |
+
INFO:root:Average loss=0.2759468169548573
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2849 | LCC = 0.8237 | SRCC = 0.8256 ]
|
| 77 |
+
INFO:root:Average loss=0.26103865756438327
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3207 | LCC = 0.8210 | SRCC = 0.8245 ]
|
| 80 |
+
INFO:root:Average loss=0.25575430328265214
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2863 | LCC = 0.8270 | SRCC = 0.8277 ]
|
| 83 |
+
INFO:root:Average loss=0.2723673280232992
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2872 | LCC = 0.8306 | SRCC = 0.8307 ]
|
| 86 |
+
INFO:root:Average loss=0.2772990440328916
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3110 | LCC = 0.8282 | SRCC = 0.8293 ]
|
| 89 |
+
INFO:root:Average loss=0.2672784169897055
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3508 | LCC = 0.8232 | SRCC = 0.8269 ]
|
| 92 |
+
INFO:root:Average loss=0.26590201277763414
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3083 | LCC = 0.8316 | SRCC = 0.8308 ]
|
| 95 |
+
INFO:root:Average loss=0.26222832921223765
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3052 | LCC = 0.8275 | SRCC = 0.8314 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2806 | LCC = 0.8491 | SRCC = 0.8478 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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_base_bvcc_finetuned_1epoch/layer10/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9d860d9f3c403d48670dbd17f4af8b7e058e0f4b0886c2ae97eca1eab96414b5
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:89da6ce2a9ba23d2a7e0d6f4421fe94296eacccb5f70ae333d4d5c2f56a3c620
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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.6074026456246009
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5345 | LCC = 0.7236 | SRCC = 0.7236 ]
|
| 8 |
+
INFO:root:Average loss=0.40795303384462994
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.6051 | LCC = 0.7298 | SRCC = 0.7651 ]
|
| 11 |
+
INFO:root:Average loss=0.4146918862676009
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3776 | LCC = 0.7675 | SRCC = 0.7758 ]
|
| 14 |
+
INFO:root:Average loss=0.3678717937989113
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3681 | LCC = 0.7764 | SRCC = 0.7816 ]
|
| 17 |
+
INFO:root:Average loss=0.35989341120689344
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.4203 | LCC = 0.7881 | SRCC = 0.7889 ]
|
| 20 |
+
INFO:root:Average loss=0.3471070033235428
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.4267 | LCC = 0.7826 | SRCC = 0.7925 ]
|
| 23 |
+
INFO:root:Average loss=0.3475740718153807
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3774 | LCC = 0.7944 | SRCC = 0.7979 ]
|
| 26 |
+
INFO:root:Average loss=0.3508263612404848
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3217 | LCC = 0.7878 | SRCC = 0.7992 ]
|
| 29 |
+
INFO:root:Average loss=0.29930401880007523
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3527 | LCC = 0.8042 | SRCC = 0.8045 ]
|
| 32 |
+
INFO:root:Average loss=0.3052817672873155
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3255 | LCC = 0.8051 | SRCC = 0.8069 ]
|
| 35 |
+
INFO:root:Average loss=0.31258848481453383
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3281 | LCC = 0.8078 | SRCC = 0.8107 ]
|
| 38 |
+
INFO:root:Average loss=0.3130040663557175
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3402 | LCC = 0.8110 | SRCC = 0.8119 ]
|
| 41 |
+
INFO:root:Average loss=0.3051484076258464
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3238 | LCC = 0.8137 | SRCC = 0.8136 ]
|
| 44 |
+
INFO:root:Average loss=0.2829606403142978
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3048 | LCC = 0.8171 | SRCC = 0.8172 ]
|
| 47 |
+
INFO:root:Average loss=0.29059502940911514
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2993 | LCC = 0.8183 | SRCC = 0.8187 ]
|
| 50 |
+
INFO:root:Average loss=0.2894028942936506
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3329 | LCC = 0.8189 | SRCC = 0.8209 ]
|
| 53 |
+
INFO:root:Average loss=0.28993573975868714
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3131 | LCC = 0.8155 | SRCC = 0.8177 ]
|
| 56 |
+
INFO:root:Average loss=0.3032311447537862
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3310 | LCC = 0.8186 | SRCC = 0.8225 ]
|
| 59 |
+
INFO:root:Average loss=0.2717400250526575
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3201 | LCC = 0.8210 | SRCC = 0.8209 ]
|
| 62 |
+
INFO:root:Average loss=0.2961716319505985
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3170 | LCC = 0.8213 | SRCC = 0.8207 ]
|
| 65 |
+
INFO:root:Average loss=0.30499528692318845
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3340 | LCC = 0.8228 | SRCC = 0.8212 ]
|
| 68 |
+
INFO:root:Average loss=0.27432053631696945
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3545 | LCC = 0.8208 | SRCC = 0.8224 ]
|
| 71 |
+
INFO:root:Average loss=0.27994152655204135
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3241 | LCC = 0.8251 | SRCC = 0.8243 ]
|
| 74 |
+
INFO:root:Average loss=0.26268342767770475
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3406 | LCC = 0.8240 | SRCC = 0.8251 ]
|
| 77 |
+
INFO:root:Average loss=0.2586102655682808
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2932 | LCC = 0.8276 | SRCC = 0.8269 ]
|
| 80 |
+
INFO:root:Average loss=0.2711377218365669
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3267 | LCC = 0.8289 | SRCC = 0.8272 ]
|
| 83 |
+
INFO:root:Average loss=0.25820626681431746
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2837 | LCC = 0.8290 | SRCC = 0.8287 ]
|
| 86 |
+
INFO:root:Average loss=0.26998501958755344
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3382 | LCC = 0.8252 | SRCC = 0.8273 ]
|
| 89 |
+
INFO:root:Average loss=0.25168586503236723
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2823 | LCC = 0.8258 | SRCC = 0.8267 ]
|
| 92 |
+
INFO:root:Average loss=0.25991330525049794
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3235 | LCC = 0.8293 | SRCC = 0.8282 ]
|
| 95 |
+
INFO:root:Average loss=0.26805608776899487
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3083 | LCC = 0.8309 | SRCC = 0.8311 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2685 | LCC = 0.8476 | SRCC = 0.8455 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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_base_bvcc_finetuned_1epoch/layer10/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6586bbec66a08b780361caee691b4106158753070c36826aefbf55ef00af1e3b
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b307556514ede366dfa7056d83cb3bd5e71a3a69cac0b088e08f7eb624482378
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer10/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.5919525225957235
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4319 | LCC = 0.7196 | SRCC = 0.7269 ]
|
| 8 |
+
INFO:root:Average loss=0.3931235648118533
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4207 | LCC = 0.7632 | SRCC = 0.7690 ]
|
| 11 |
+
INFO:root:Average loss=0.3702609174144574
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.4174 | LCC = 0.7676 | SRCC = 0.7700 ]
|
| 14 |
+
INFO:root:Average loss=0.39132164132136565
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3678 | LCC = 0.7762 | SRCC = 0.7778 ]
|
| 17 |
+
INFO:root:Average loss=0.3711143535299179
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3688 | LCC = 0.7859 | SRCC = 0.7861 ]
|
| 20 |
+
INFO:root:Average loss=0.32915401248595655
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3608 | LCC = 0.7899 | SRCC = 0.7915 ]
|
| 23 |
+
INFO:root:Average loss=0.33503451981605625
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3533 | LCC = 0.7919 | SRCC = 0.7947 ]
|
| 26 |
+
INFO:root:Average loss=0.35452189296483994
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3331 | LCC = 0.8011 | SRCC = 0.8017 ]
|
| 29 |
+
INFO:root:Average loss=0.3177190146003014
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.4371 | LCC = 0.7977 | SRCC = 0.8012 ]
|
| 32 |
+
INFO:root:Average loss=0.3301977371940246
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3284 | LCC = 0.8029 | SRCC = 0.8029 ]
|
| 35 |
+
INFO:root:Average loss=0.32799746287174714
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3176 | LCC = 0.8065 | SRCC = 0.8078 ]
|
| 38 |
+
INFO:root:Average loss=0.3009518109835111
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3393 | LCC = 0.8084 | SRCC = 0.8070 ]
|
| 41 |
+
INFO:root:Average loss=0.30156356688493335
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3293 | LCC = 0.8052 | SRCC = 0.8070 ]
|
| 44 |
+
INFO:root:Average loss=0.31853661972742814
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3287 | LCC = 0.8120 | SRCC = 0.8117 ]
|
| 47 |
+
INFO:root:Average loss=0.29982112500912106
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3996 | LCC = 0.8077 | SRCC = 0.8099 ]
|
| 50 |
+
INFO:root:Average loss=0.3019583529004684
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3620 | LCC = 0.8151 | SRCC = 0.8158 ]
|
| 53 |
+
INFO:root:Average loss=0.2804290268283624
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3148 | LCC = 0.8142 | SRCC = 0.8143 ]
|
| 56 |
+
INFO:root:Average loss=0.27405899189985716
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3193 | LCC = 0.8188 | SRCC = 0.8171 ]
|
| 59 |
+
INFO:root:Average loss=0.2996655435134203
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3444 | LCC = 0.8211 | SRCC = 0.8193 ]
|
| 62 |
+
INFO:root:Average loss=0.2868417201515956
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3405 | LCC = 0.8231 | SRCC = 0.8210 ]
|
| 65 |
+
INFO:root:Average loss=0.2829140644425001
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3079 | LCC = 0.8099 | SRCC = 0.8210 ]
|
| 68 |
+
INFO:root:Average loss=0.2662664563992085
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3098 | LCC = 0.8201 | SRCC = 0.8206 ]
|
| 71 |
+
INFO:root:Average loss=0.277060608451183
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2927 | LCC = 0.8234 | SRCC = 0.8237 ]
|
| 74 |
+
INFO:root:Average loss=0.275130726205997
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3379 | LCC = 0.8210 | SRCC = 0.8213 ]
|
| 77 |
+
INFO:root:Average loss=0.27564884340151763
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2925 | LCC = 0.8256 | SRCC = 0.8246 ]
|
| 80 |
+
INFO:root:Average loss=0.264566234480112
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3564 | LCC = 0.8263 | SRCC = 0.8251 ]
|
| 83 |
+
INFO:root:Average loss=0.2591870858883246
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3351 | LCC = 0.8287 | SRCC = 0.8251 ]
|
| 86 |
+
INFO:root:Average loss=0.2626095484846678
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3118 | LCC = 0.8280 | SRCC = 0.8270 ]
|
| 89 |
+
INFO:root:Average loss=0.26458298644194234
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3027 | LCC = 0.8279 | SRCC = 0.8266 ]
|
| 92 |
+
INFO:root:Average loss=0.25217374796286607
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2870 | LCC = 0.8287 | SRCC = 0.8282 ]
|
| 95 |
+
INFO:root:Average loss=0.2601147655111093
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2859 | LCC = 0.8285 | SRCC = 0.8283 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2592 | LCC = 0.8478 | SRCC = 0.8481 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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_base_bvcc_finetuned_1epoch/layer11/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:156f69c399c4893e577902557dd134bccd75a0faa5d90e4ddcf1a2be9f90e06b
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0e75a3ff80515a5e9fa48cee3a99b9bf56301b72ac7628a5df52ca6757111f15
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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.6729580377921079
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.6144 | LCC = 0.6437 | SRCC = 0.6433 ]
|
| 8 |
+
INFO:root:Average loss=0.4392205820633815
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.5890 | LCC = 0.6986 | SRCC = 0.7416 ]
|
| 11 |
+
INFO:root:Average loss=0.42539280347335034
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.4591 | LCC = 0.7428 | SRCC = 0.7544 ]
|
| 14 |
+
INFO:root:Average loss=0.4404584815104802
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.4064 | LCC = 0.7589 | SRCC = 0.7627 ]
|
| 17 |
+
INFO:root:Average loss=0.39679989505272645
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3677 | LCC = 0.7607 | SRCC = 0.7660 ]
|
| 20 |
+
INFO:root:Average loss=0.35056606641946697
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.4482 | LCC = 0.7583 | SRCC = 0.7716 ]
|
| 23 |
+
INFO:root:Average loss=0.3767879463923283
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3750 | LCC = 0.7693 | SRCC = 0.7766 ]
|
| 26 |
+
INFO:root:Average loss=0.3796848373917433
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3820 | LCC = 0.7779 | SRCC = 0.7826 ]
|
| 29 |
+
INFO:root:Average loss=0.3379638451032149
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3881 | LCC = 0.7753 | SRCC = 0.7808 ]
|
| 32 |
+
INFO:root:Average loss=0.3764944424231847
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.4350 | LCC = 0.7812 | SRCC = 0.7878 ]
|
| 35 |
+
INFO:root:Average loss=0.33237109180444324
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.4639 | LCC = 0.7711 | SRCC = 0.7868 ]
|
| 38 |
+
INFO:root:Average loss=0.32199583603785586
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3349 | LCC = 0.7930 | SRCC = 0.7950 ]
|
| 41 |
+
INFO:root:Average loss=0.32168945345358974
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3347 | LCC = 0.7957 | SRCC = 0.7978 ]
|
| 44 |
+
INFO:root:Average loss=0.33441059310466814
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3615 | LCC = 0.7974 | SRCC = 0.7990 ]
|
| 47 |
+
INFO:root:Average loss=0.32213599998981524
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3278 | LCC = 0.7976 | SRCC = 0.8003 ]
|
| 50 |
+
INFO:root:Average loss=0.3144732795846768
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3469 | LCC = 0.8020 | SRCC = 0.8024 ]
|
| 53 |
+
INFO:root:Average loss=0.30663015311345077
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3637 | LCC = 0.7976 | SRCC = 0.8008 ]
|
| 56 |
+
INFO:root:Average loss=0.29915269101277375
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3612 | LCC = 0.8025 | SRCC = 0.8045 ]
|
| 59 |
+
INFO:root:Average loss=0.2893740036166631
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3567 | LCC = 0.8006 | SRCC = 0.8061 ]
|
| 62 |
+
INFO:root:Average loss=0.311808311404326
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3223 | LCC = 0.8037 | SRCC = 0.8075 ]
|
| 65 |
+
INFO:root:Average loss=0.29704856891662645
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3461 | LCC = 0.8105 | SRCC = 0.8098 ]
|
| 68 |
+
INFO:root:Average loss=0.2921784422718562
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3170 | LCC = 0.8082 | SRCC = 0.8084 ]
|
| 71 |
+
INFO:root:Average loss=0.30228491605092317
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.4087 | LCC = 0.8081 | SRCC = 0.8082 ]
|
| 74 |
+
INFO:root:Average loss=0.2978213614760301
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.4036 | LCC = 0.7952 | SRCC = 0.8056 ]
|
| 77 |
+
INFO:root:Average loss=0.33107572125318724
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3237 | LCC = 0.8100 | SRCC = 0.8101 ]
|
| 80 |
+
INFO:root:Average loss=0.29745176358100694
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3308 | LCC = 0.8117 | SRCC = 0.8119 ]
|
| 83 |
+
INFO:root:Average loss=0.28526496486021924
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3792 | LCC = 0.8086 | SRCC = 0.8102 ]
|
| 86 |
+
INFO:root:Average loss=0.30388311869823015
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3872 | LCC = 0.8080 | SRCC = 0.8119 ]
|
| 89 |
+
INFO:root:Average loss=0.2939731801549594
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3466 | LCC = 0.8124 | SRCC = 0.8131 ]
|
| 92 |
+
INFO:root:Average loss=0.2995765264599751
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3417 | LCC = 0.8034 | SRCC = 0.8077 ]
|
| 95 |
+
INFO:root:Average loss=0.3072073830243869
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3827 | LCC = 0.8101 | SRCC = 0.8120 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3347 | LCC = 0.8324 | SRCC = 0.8372 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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_base_bvcc_finetuned_1epoch/layer11/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:929c145cc02debf9c0538911e79d5d5bde07d8f418d1e5ee745398b2ed48e2c2
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ab70190b6c141dc788dcc38b53d068e6e90332a665c68b428cca83f8487cbb30
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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.6581237021929178
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.6564 | LCC = 0.6689 | SRCC = 0.6788 ]
|
| 8 |
+
INFO:root:Average loss=0.42959490991555727
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4299 | LCC = 0.7417 | SRCC = 0.7464 ]
|
| 11 |
+
INFO:root:Average loss=0.37951474617689085
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.4261 | LCC = 0.7583 | SRCC = 0.7603 ]
|
| 14 |
+
INFO:root:Average loss=0.39183056889436185
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3669 | LCC = 0.7610 | SRCC = 0.7696 ]
|
| 17 |
+
INFO:root:Average loss=0.4422425774809642
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.4534 | LCC = 0.7616 | SRCC = 0.7717 ]
|
| 20 |
+
INFO:root:Average loss=0.3932936424628282
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.6588 | LCC = 0.7383 | SRCC = 0.7689 ]
|
| 23 |
+
INFO:root:Average loss=0.3543844696802971
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3613 | LCC = 0.7733 | SRCC = 0.7796 ]
|
| 26 |
+
INFO:root:Average loss=0.36080430161494476
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3836 | LCC = 0.7860 | SRCC = 0.7873 ]
|
| 29 |
+
INFO:root:Average loss=0.38030731773529297
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.4231 | LCC = 0.7775 | SRCC = 0.7821 ]
|
| 32 |
+
INFO:root:Average loss=0.3319390423786946
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3978 | LCC = 0.7770 | SRCC = 0.7874 ]
|
| 35 |
+
INFO:root:Average loss=0.3409799784421921
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.4475 | LCC = 0.7822 | SRCC = 0.7921 ]
|
| 38 |
+
INFO:root:Average loss=0.3385501202100363
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.5145 | LCC = 0.7766 | SRCC = 0.7912 ]
|
| 41 |
+
INFO:root:Average loss=0.34339119092776227
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.4081 | LCC = 0.7830 | SRCC = 0.7937 ]
|
| 44 |
+
INFO:root:Average loss=0.3401635623513124
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3651 | LCC = 0.7977 | SRCC = 0.7993 ]
|
| 47 |
+
INFO:root:Average loss=0.31414138640348727
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.4214 | LCC = 0.7905 | SRCC = 0.7952 ]
|
| 50 |
+
INFO:root:Average loss=0.3529744694630305
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3782 | LCC = 0.8015 | SRCC = 0.8014 ]
|
| 53 |
+
INFO:root:Average loss=0.31871499369541806
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3447 | LCC = 0.7906 | SRCC = 0.7999 ]
|
| 56 |
+
INFO:root:Average loss=0.29900134335725737
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3306 | LCC = 0.8037 | SRCC = 0.8041 ]
|
| 59 |
+
INFO:root:Average loss=0.3000396258938007
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3486 | LCC = 0.8065 | SRCC = 0.8069 ]
|
| 62 |
+
INFO:root:Average loss=0.30084676047166187
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3331 | LCC = 0.8064 | SRCC = 0.8087 ]
|
| 65 |
+
INFO:root:Average loss=0.3170011144800064
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3987 | LCC = 0.7968 | SRCC = 0.8060 ]
|
| 68 |
+
INFO:root:Average loss=0.3147989662411885
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3197 | LCC = 0.8078 | SRCC = 0.8096 ]
|
| 71 |
+
INFO:root:Average loss=0.2965874480895507
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3131 | LCC = 0.8099 | SRCC = 0.8112 ]
|
| 74 |
+
INFO:root:Average loss=0.32708879101734895
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3991 | LCC = 0.8095 | SRCC = 0.8090 ]
|
| 77 |
+
INFO:root:Average loss=0.3183735649173076
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3126 | LCC = 0.8075 | SRCC = 0.8103 ]
|
| 80 |
+
INFO:root:Average loss=0.31792674079919475
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3612 | LCC = 0.8132 | SRCC = 0.8122 ]
|
| 83 |
+
INFO:root:Average loss=0.2930086820553511
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.4721 | LCC = 0.7961 | SRCC = 0.8079 ]
|
| 86 |
+
INFO:root:Average loss=0.28792792768814623
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2961 | LCC = 0.8119 | SRCC = 0.8141 ]
|
| 89 |
+
INFO:root:Average loss=0.30539400722735965
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3561 | LCC = 0.8113 | SRCC = 0.8112 ]
|
| 92 |
+
INFO:root:Average loss=0.30708794162059444
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.4237 | LCC = 0.8134 | SRCC = 0.8139 ]
|
| 95 |
+
INFO:root:Average loss=0.3381939790187738
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3813 | LCC = 0.8095 | SRCC = 0.8093 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.4106 | LCC = 0.8301 | SRCC = 0.8366 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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_base_bvcc_finetuned_1epoch/layer11/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:822f4fa1222f31160b4d2d84bc98404560288f94f7045582125a5bacfd9a26bb
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b181cd74d514ee4d068470f01e23bfce0051fe5c22da7dcee1501ef313356b8
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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.6874579221774371
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5044 | LCC = 0.6915 | SRCC = 0.6913 ]
|
| 8 |
+
INFO:root:Average loss=0.4680313211985124
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4324 | LCC = 0.7318 | SRCC = 0.7362 ]
|
| 11 |
+
INFO:root:Average loss=0.37523861649708873
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.4084 | LCC = 0.7505 | SRCC = 0.7559 ]
|
| 14 |
+
INFO:root:Average loss=0.3899931791119086
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3897 | LCC = 0.7627 | SRCC = 0.7641 ]
|
| 17 |
+
INFO:root:Average loss=0.3557039323525551
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.4442 | LCC = 0.7627 | SRCC = 0.7732 ]
|
| 20 |
+
INFO:root:Average loss=0.3772805075232799
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3816 | LCC = 0.7775 | SRCC = 0.7786 ]
|
| 23 |
+
INFO:root:Average loss=0.3755926938775258
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3889 | LCC = 0.7840 | SRCC = 0.7837 ]
|
| 26 |
+
INFO:root:Average loss=0.33382165279143894
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.4150 | LCC = 0.7837 | SRCC = 0.7876 ]
|
| 29 |
+
INFO:root:Average loss=0.35614499105856967
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3999 | LCC = 0.7898 | SRCC = 0.7896 ]
|
| 32 |
+
INFO:root:Average loss=0.31742671514168763
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3383 | LCC = 0.7871 | SRCC = 0.7906 ]
|
| 35 |
+
INFO:root:Average loss=0.3147903044636433
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3937 | LCC = 0.7951 | SRCC = 0.7972 ]
|
| 38 |
+
INFO:root:Average loss=0.32551616258346117
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3781 | LCC = 0.7992 | SRCC = 0.7993 ]
|
| 41 |
+
INFO:root:Average loss=0.34049213257355565
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.4585 | LCC = 0.7973 | SRCC = 0.8017 ]
|
| 44 |
+
INFO:root:Average loss=0.3466266693595128
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3762 | LCC = 0.7966 | SRCC = 0.7947 ]
|
| 47 |
+
INFO:root:Average loss=0.30925640158164197
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.4061 | LCC = 0.8015 | SRCC = 0.8014 ]
|
| 50 |
+
INFO:root:Average loss=0.30770480995758986
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3715 | LCC = 0.8040 | SRCC = 0.8031 ]
|
| 53 |
+
INFO:root:Average loss=0.29292874401196456
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3240 | LCC = 0.8037 | SRCC = 0.8033 ]
|
| 56 |
+
INFO:root:Average loss=0.2952551343120061
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3670 | LCC = 0.8071 | SRCC = 0.8074 ]
|
| 59 |
+
INFO:root:Average loss=0.33644014539626926
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3722 | LCC = 0.8059 | SRCC = 0.8057 ]
|
| 62 |
+
INFO:root:Average loss=0.32253950394881076
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3672 | LCC = 0.8078 | SRCC = 0.8087 ]
|
| 65 |
+
INFO:root:Average loss=0.30008579389407086
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3328 | LCC = 0.8028 | SRCC = 0.8031 ]
|
| 68 |
+
INFO:root:Average loss=0.29782283707306934
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3807 | LCC = 0.8116 | SRCC = 0.8104 ]
|
| 71 |
+
INFO:root:Average loss=0.29301549819035405
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.4340 | LCC = 0.8114 | SRCC = 0.8106 ]
|
| 74 |
+
INFO:root:Average loss=0.3072131071717311
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3356 | LCC = 0.8097 | SRCC = 0.8093 ]
|
| 77 |
+
INFO:root:Average loss=0.28786266098419827
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3029 | LCC = 0.8053 | SRCC = 0.8075 ]
|
| 80 |
+
INFO:root:Average loss=0.29310321769653225
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3536 | LCC = 0.8119 | SRCC = 0.8122 ]
|
| 83 |
+
INFO:root:Average loss=0.28970870776818347
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3620 | LCC = 0.8040 | SRCC = 0.8074 ]
|
| 86 |
+
INFO:root:Average loss=0.27608862633888537
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3451 | LCC = 0.8156 | SRCC = 0.8139 ]
|
| 89 |
+
INFO:root:Average loss=0.27924639750749636
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3395 | LCC = 0.8153 | SRCC = 0.8148 ]
|
| 92 |
+
INFO:root:Average loss=0.2864933799092586
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3340 | LCC = 0.8154 | SRCC = 0.8129 ]
|
| 95 |
+
INFO:root:Average loss=0.2967428853496527
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3296 | LCC = 0.8172 | SRCC = 0.8164 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3042 | LCC = 0.8362 | SRCC = 0.8360 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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_base_bvcc_finetuned_1epoch/layer11/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31ebfbf93784f981fb56f1903b11bf73d163febc6411afda3ebd1b5c42b83dfd
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8b8c4057199440db0745706bdbed3a1989bac63d381646c6ff5be876468bdf76
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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.696067886474805
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5179 | LCC = 0.6556 | SRCC = 0.6590 ]
|
| 8 |
+
INFO:root:Average loss=0.44739482761957705
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.5517 | LCC = 0.7096 | SRCC = 0.7462 ]
|
| 11 |
+
INFO:root:Average loss=0.3830511042704949
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3875 | LCC = 0.7497 | SRCC = 0.7602 ]
|
| 14 |
+
INFO:root:Average loss=0.38545174476427907
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3973 | LCC = 0.7579 | SRCC = 0.7660 ]
|
| 17 |
+
INFO:root:Average loss=0.38210188731169087
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3735 | LCC = 0.7675 | SRCC = 0.7702 ]
|
| 20 |
+
INFO:root:Average loss=0.4005351104797461
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3601 | LCC = 0.7682 | SRCC = 0.7728 ]
|
| 23 |
+
INFO:root:Average loss=0.33564635003224397
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3727 | LCC = 0.7756 | SRCC = 0.7815 ]
|
| 26 |
+
INFO:root:Average loss=0.35139848330082035
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3497 | LCC = 0.7824 | SRCC = 0.7838 ]
|
| 29 |
+
INFO:root:Average loss=0.3481192625103853
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3655 | LCC = 0.7852 | SRCC = 0.7878 ]
|
| 32 |
+
INFO:root:Average loss=0.34001534183820087
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3633 | LCC = 0.7868 | SRCC = 0.7874 ]
|
| 35 |
+
INFO:root:Average loss=0.3134442233504393
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3377 | LCC = 0.7893 | SRCC = 0.7922 ]
|
| 38 |
+
INFO:root:Average loss=0.3428114320223148
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3650 | LCC = 0.7949 | SRCC = 0.7955 ]
|
| 41 |
+
INFO:root:Average loss=0.3338470262212631
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.4090 | LCC = 0.7799 | SRCC = 0.7894 ]
|
| 44 |
+
INFO:root:Average loss=0.3234367775611388
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3352 | LCC = 0.7934 | SRCC = 0.7966 ]
|
| 47 |
+
INFO:root:Average loss=0.32435524520965725
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3531 | LCC = 0.7989 | SRCC = 0.8006 ]
|
| 50 |
+
INFO:root:Average loss=0.36163984678494626
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.4671 | LCC = 0.7902 | SRCC = 0.7992 ]
|
| 53 |
+
INFO:root:Average loss=0.32743537655243504
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3657 | LCC = 0.7930 | SRCC = 0.7970 ]
|
| 56 |
+
INFO:root:Average loss=0.320269026244298
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3388 | LCC = 0.8004 | SRCC = 0.8021 ]
|
| 59 |
+
INFO:root:Average loss=0.3269584895326541
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3997 | LCC = 0.8004 | SRCC = 0.8007 ]
|
| 62 |
+
INFO:root:Average loss=0.3281559944152832
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3254 | LCC = 0.7997 | SRCC = 0.8033 ]
|
| 65 |
+
INFO:root:Average loss=0.31543307598584736
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3537 | LCC = 0.8034 | SRCC = 0.8037 ]
|
| 68 |
+
INFO:root:Average loss=0.2931396451133948
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3467 | LCC = 0.7998 | SRCC = 0.8057 ]
|
| 71 |
+
INFO:root:Average loss=0.3020174054381175
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3369 | LCC = 0.8035 | SRCC = 0.8027 ]
|
| 74 |
+
INFO:root:Average loss=0.2944817732159908
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3415 | LCC = 0.8067 | SRCC = 0.8068 ]
|
| 77 |
+
INFO:root:Average loss=0.2979569958570676
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3795 | LCC = 0.7986 | SRCC = 0.8057 ]
|
| 80 |
+
INFO:root:Average loss=0.2853924439121515
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3471 | LCC = 0.8087 | SRCC = 0.8097 ]
|
| 83 |
+
INFO:root:Average loss=0.292974207454767
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3312 | LCC = 0.8116 | SRCC = 0.8094 ]
|
| 86 |
+
INFO:root:Average loss=0.3221439260702867
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3133 | LCC = 0.8106 | SRCC = 0.8108 ]
|
| 89 |
+
INFO:root:Average loss=0.29090672654983324
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3578 | LCC = 0.8103 | SRCC = 0.8110 ]
|
| 92 |
+
INFO:root:Average loss=0.2899026366380545
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3106 | LCC = 0.8123 | SRCC = 0.8120 ]
|
| 95 |
+
INFO:root:Average loss=0.27947496011471135
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3250 | LCC = 0.8126 | SRCC = 0.8104 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2931 | LCC = 0.8325 | SRCC = 0.8324 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/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_base_bvcc_finetuned_1epoch/layer11/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ae7fabacf8e288a5b8d4bac4475ba91aee64a11b3e63d26d81b6b1e67e9a297d
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_1epoch/layer11/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7dc7dcb0f4f70ce33911cd87e0d22e8a006d91d0c765689e1557ed97f2549b2b
|
| 3 |
+
size 2749638
|