Upload SSL Layer MOS checkpoints (part 8)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run4/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run4/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run0/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run0/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run0/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run0/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run1/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run1/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run1/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run1/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run2/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run4/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run4/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run0/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run0/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run0/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run0/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run1/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run1/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run1/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run1/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run2/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run4/model_best.pt +3 -0
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e6b50c9387dcbe58e205f9a57677b09388adccb43bdcedee1f59d0478f4c8355
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b99ff50d28a48dacfe0bf472bec9104893ed3d08d396214babc54bebba8b268b
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/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.3602916034750449
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3017 | LCC = 0.8297 | SRCC = 0.8342 ]
|
| 8 |
+
INFO:root:Average loss=0.22860305699018332
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2704 | LCC = 0.8552 | SRCC = 0.8523 ]
|
| 11 |
+
INFO:root:Average loss=0.2008502225463207
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2308 | LCC = 0.8624 | SRCC = 0.8625 ]
|
| 14 |
+
INFO:root:Average loss=0.185303858648508
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2760 | LCC = 0.8650 | SRCC = 0.8637 ]
|
| 17 |
+
INFO:root:Average loss=0.17254548395673433
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2495 | LCC = 0.8698 | SRCC = 0.8676 ]
|
| 20 |
+
INFO:root:Average loss=0.15667101721733045
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2468 | LCC = 0.8667 | SRCC = 0.8644 ]
|
| 23 |
+
INFO:root:Average loss=0.1475692986486814
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2355 | LCC = 0.8645 | SRCC = 0.8632 ]
|
| 26 |
+
INFO:root:Average loss=0.13897179391903755
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2490 | LCC = 0.8602 | SRCC = 0.8608 ]
|
| 29 |
+
INFO:root:Average loss=0.1250226007631192
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2292 | LCC = 0.8599 | SRCC = 0.8590 ]
|
| 32 |
+
INFO:root:Average loss=0.1178060746154724
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2492 | LCC = 0.8604 | SRCC = 0.8591 ]
|
| 35 |
+
INFO:root:Average loss=0.10731051575679046
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2352 | LCC = 0.8585 | SRCC = 0.8561 ]
|
| 38 |
+
INFO:root:Average loss=0.10309349201046504
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2484 | LCC = 0.8597 | SRCC = 0.8569 ]
|
| 41 |
+
INFO:root:Average loss=0.09537047563264003
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2379 | LCC = 0.8544 | SRCC = 0.8522 ]
|
| 44 |
+
INFO:root:Average loss=0.08652413985094963
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2674 | LCC = 0.8531 | SRCC = 0.8496 ]
|
| 47 |
+
INFO:root:Average loss=0.0837886471969959
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2418 | LCC = 0.8534 | SRCC = 0.8531 ]
|
| 50 |
+
INFO:root:Average loss=0.07590657128737523
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2412 | LCC = 0.8523 | SRCC = 0.8515 ]
|
| 53 |
+
INFO:root:Average loss=0.07344822246485795
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2441 | LCC = 0.8499 | SRCC = 0.8483 ]
|
| 56 |
+
INFO:root:Average loss=0.07031746958501828
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2497 | LCC = 0.8503 | SRCC = 0.8488 ]
|
| 59 |
+
INFO:root:Average loss=0.0652227365435698
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2388 | LCC = 0.8523 | SRCC = 0.8509 ]
|
| 62 |
+
INFO:root:Average loss=0.06416319434841473
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2686 | LCC = 0.8484 | SRCC = 0.8486 ]
|
| 65 |
+
INFO:root:Average loss=0.06138015863222954
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2499 | LCC = 0.8480 | SRCC = 0.8456 ]
|
| 68 |
+
INFO:root:Average loss=0.05661991114417712
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2570 | LCC = 0.8495 | SRCC = 0.8489 ]
|
| 71 |
+
INFO:root:Average loss=0.05485928946962723
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2561 | LCC = 0.8467 | SRCC = 0.8468 ]
|
| 74 |
+
INFO:root:Average loss=0.05111394521708672
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2419 | LCC = 0.8487 | SRCC = 0.8462 ]
|
| 77 |
+
INFO:root:Average loss=0.05058705133314316
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2489 | LCC = 0.8469 | SRCC = 0.8445 ]
|
| 80 |
+
INFO:root:Average loss=0.04935907319379158
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2400 | LCC = 0.8457 | SRCC = 0.8447 ]
|
| 83 |
+
INFO:root:Average loss=0.04663730957187139
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2469 | LCC = 0.8478 | SRCC = 0.8456 ]
|
| 86 |
+
INFO:root:Average loss=0.04633009780007295
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2476 | LCC = 0.8467 | SRCC = 0.8461 ]
|
| 89 |
+
INFO:root:Average loss=0.04485997715248512
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2528 | LCC = 0.8471 | SRCC = 0.8462 ]
|
| 92 |
+
INFO:root:Average loss=0.04893341567367315
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2455 | LCC = 0.8453 | SRCC = 0.8441 ]
|
| 95 |
+
INFO:root:Average loss=0.04172439204576688
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2377 | LCC = 0.8469 | SRCC = 0.8457 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2426 | LCC = 0.8652 | SRCC = 0.8651 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/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_8epoch/layer5/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:69884306410746ecb8677bfefed6eb0b64c36c6dc02b6daca2cad34720312c19
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4bc767021def3f224f03464651aad93bce7804f579a71f1711116c8bafb7ab1a
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/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.362190163288361
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3009 | LCC = 0.8301 | SRCC = 0.8381 ]
|
| 8 |
+
INFO:root:Average loss=0.2284203271071116
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2673 | LCC = 0.8539 | SRCC = 0.8531 ]
|
| 11 |
+
INFO:root:Average loss=0.20273739557999831
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2613 | LCC = 0.8658 | SRCC = 0.8636 ]
|
| 14 |
+
INFO:root:Average loss=0.18265455999435523
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2410 | LCC = 0.8670 | SRCC = 0.8649 ]
|
| 17 |
+
INFO:root:Average loss=0.17725740506862983
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2656 | LCC = 0.8689 | SRCC = 0.8669 ]
|
| 20 |
+
INFO:root:Average loss=0.1582477321036351
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2515 | LCC = 0.8697 | SRCC = 0.8684 ]
|
| 23 |
+
INFO:root:Average loss=0.14821337889402342
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2149 | LCC = 0.8709 | SRCC = 0.8709 ]
|
| 26 |
+
INFO:root:Average loss=0.14272961555383143
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2318 | LCC = 0.8722 | SRCC = 0.8715 ]
|
| 29 |
+
INFO:root:Average loss=0.12586095403784361
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2161 | LCC = 0.8669 | SRCC = 0.8674 ]
|
| 32 |
+
INFO:root:Average loss=0.12062422826122014
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2265 | LCC = 0.8707 | SRCC = 0.8704 ]
|
| 35 |
+
INFO:root:Average loss=0.11422893978082217
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2333 | LCC = 0.8650 | SRCC = 0.8649 ]
|
| 38 |
+
INFO:root:Average loss=0.10612992885021064
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2235 | LCC = 0.8654 | SRCC = 0.8653 ]
|
| 41 |
+
INFO:root:Average loss=0.09691693519170468
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2206 | LCC = 0.8645 | SRCC = 0.8653 ]
|
| 44 |
+
INFO:root:Average loss=0.09305107727264747
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2303 | LCC = 0.8633 | SRCC = 0.8627 ]
|
| 47 |
+
INFO:root:Average loss=0.0835542368869751
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2274 | LCC = 0.8627 | SRCC = 0.8636 ]
|
| 50 |
+
INFO:root:Average loss=0.08460547426381172
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2450 | LCC = 0.8617 | SRCC = 0.8615 ]
|
| 53 |
+
INFO:root:Average loss=0.07684859504493383
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2259 | LCC = 0.8592 | SRCC = 0.8591 ]
|
| 56 |
+
INFO:root:Average loss=0.07778260527321926
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2414 | LCC = 0.8588 | SRCC = 0.8589 ]
|
| 59 |
+
INFO:root:Average loss=0.06990714949101974
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2353 | LCC = 0.8605 | SRCC = 0.8596 ]
|
| 62 |
+
INFO:root:Average loss=0.06567355474600425
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2199 | LCC = 0.8611 | SRCC = 0.8613 ]
|
| 65 |
+
INFO:root:Average loss=0.06059241590973658
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2355 | LCC = 0.8560 | SRCC = 0.8553 ]
|
| 68 |
+
INFO:root:Average loss=0.058565773404179476
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2288 | LCC = 0.8588 | SRCC = 0.8592 ]
|
| 71 |
+
INFO:root:Average loss=0.05620270083921078
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2361 | LCC = 0.8612 | SRCC = 0.8620 ]
|
| 74 |
+
INFO:root:Average loss=0.05402580209267445
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2293 | LCC = 0.8571 | SRCC = 0.8579 ]
|
| 77 |
+
INFO:root:Average loss=0.0521805359958074
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2434 | LCC = 0.8537 | SRCC = 0.8538 ]
|
| 80 |
+
INFO:root:Average loss=0.0494673982835733
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2347 | LCC = 0.8553 | SRCC = 0.8554 ]
|
| 83 |
+
INFO:root:Average loss=0.047855974055635624
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2351 | LCC = 0.8556 | SRCC = 0.8555 ]
|
| 86 |
+
INFO:root:Average loss=0.045182778691061035
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2422 | LCC = 0.8575 | SRCC = 0.8570 ]
|
| 89 |
+
INFO:root:Average loss=0.04461886026920416
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2312 | LCC = 0.8586 | SRCC = 0.8585 ]
|
| 92 |
+
INFO:root:Average loss=0.04863246611486643
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2319 | LCC = 0.8594 | SRCC = 0.8594 ]
|
| 95 |
+
INFO:root:Average loss=0.04162825722820484
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2430 | LCC = 0.8544 | SRCC = 0.8547 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2513 | LCC = 0.8650 | SRCC = 0.8636 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/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_8epoch/layer5/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cc0bad2deb9bb1684da3dabc3f58b3b06de9ec6bf59170afbc4d9e27cf501f80
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:532c95da86fd5cb69f939eae1706a57c03831d7bf58fbf91b39978c90f6bbfd0
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer5/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.42477776454045224
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3218 | LCC = 0.8238 | SRCC = 0.8294 ]
|
| 8 |
+
INFO:root:Average loss=0.23074349073263314
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2693 | LCC = 0.8502 | SRCC = 0.8498 ]
|
| 11 |
+
INFO:root:Average loss=0.21214958547781676
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2734 | LCC = 0.8590 | SRCC = 0.8568 ]
|
| 14 |
+
INFO:root:Average loss=0.18804034265952232
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2621 | LCC = 0.8589 | SRCC = 0.8581 ]
|
| 17 |
+
INFO:root:Average loss=0.16857376904823843
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2477 | LCC = 0.8623 | SRCC = 0.8611 ]
|
| 20 |
+
INFO:root:Average loss=0.15654911922338682
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2614 | LCC = 0.8590 | SRCC = 0.8580 ]
|
| 23 |
+
INFO:root:Average loss=0.144945139113145
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2632 | LCC = 0.8569 | SRCC = 0.8559 ]
|
| 26 |
+
INFO:root:Average loss=0.13225655763959274
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2594 | LCC = 0.8559 | SRCC = 0.8553 ]
|
| 29 |
+
INFO:root:Average loss=0.1209458742195215
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2373 | LCC = 0.8545 | SRCC = 0.8548 ]
|
| 32 |
+
INFO:root:Average loss=0.11163341435484397
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2365 | LCC = 0.8569 | SRCC = 0.8569 ]
|
| 35 |
+
INFO:root:Average loss=0.10707223711487575
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2548 | LCC = 0.8537 | SRCC = 0.8537 ]
|
| 38 |
+
INFO:root:Average loss=0.09564200382775222
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2528 | LCC = 0.8518 | SRCC = 0.8520 ]
|
| 41 |
+
INFO:root:Average loss=0.09240232288646393
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2525 | LCC = 0.8513 | SRCC = 0.8511 ]
|
| 44 |
+
INFO:root:Average loss=0.08303699680627921
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2726 | LCC = 0.8468 | SRCC = 0.8463 ]
|
| 47 |
+
INFO:root:Average loss=0.08223164812303506
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2445 | LCC = 0.8524 | SRCC = 0.8526 ]
|
| 50 |
+
INFO:root:Average loss=0.07442832212799634
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2502 | LCC = 0.8453 | SRCC = 0.8476 ]
|
| 53 |
+
INFO:root:Average loss=0.07438317820047721
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2584 | LCC = 0.8472 | SRCC = 0.8471 ]
|
| 56 |
+
INFO:root:Average loss=0.06566222432332161
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2409 | LCC = 0.8496 | SRCC = 0.8500 ]
|
| 59 |
+
INFO:root:Average loss=0.0633346375842125
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2424 | LCC = 0.8464 | SRCC = 0.8473 ]
|
| 62 |
+
INFO:root:Average loss=0.062345233507072315
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2516 | LCC = 0.8499 | SRCC = 0.8480 ]
|
| 65 |
+
INFO:root:Average loss=0.06149311549961567
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2590 | LCC = 0.8481 | SRCC = 0.8481 ]
|
| 68 |
+
INFO:root:Average loss=0.057354543883448995
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2502 | LCC = 0.8477 | SRCC = 0.8473 ]
|
| 71 |
+
INFO:root:Average loss=0.05565970307454849
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2667 | LCC = 0.8479 | SRCC = 0.8477 ]
|
| 74 |
+
INFO:root:Average loss=0.05516067046958666
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2483 | LCC = 0.8464 | SRCC = 0.8475 ]
|
| 77 |
+
INFO:root:Average loss=0.04986106446729256
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2603 | LCC = 0.8465 | SRCC = 0.8480 ]
|
| 80 |
+
INFO:root:Average loss=0.050457364425827295
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2401 | LCC = 0.8512 | SRCC = 0.8521 ]
|
| 83 |
+
INFO:root:Average loss=0.04614992647503431
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2470 | LCC = 0.8502 | SRCC = 0.8497 ]
|
| 86 |
+
INFO:root:Average loss=0.04329039013156524
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2521 | LCC = 0.8483 | SRCC = 0.8483 ]
|
| 89 |
+
INFO:root:Average loss=0.0468870301802571
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2459 | LCC = 0.8471 | SRCC = 0.8469 ]
|
| 92 |
+
INFO:root:Average loss=0.041542455864449344
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2544 | LCC = 0.8480 | SRCC = 0.8482 ]
|
| 95 |
+
INFO:root:Average loss=0.04111270451297363
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2684 | LCC = 0.8404 | SRCC = 0.8402 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2458 | LCC = 0.8641 | SRCC = 0.8660 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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_8epoch/layer6/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4ce8cbfd7b297a9bc73cf79d8557e7f9b95f747a18d6c73a5b0c34fd755f043
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9fc80e491c15e22e6525277680d7e91ea160cb9d89a4e88013c7bc5c8dafb32b
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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.37467708037449765
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2990 | LCC = 0.8281 | SRCC = 0.8343 ]
|
| 8 |
+
INFO:root:Average loss=0.2333670946267935
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3143 | LCC = 0.8528 | SRCC = 0.8531 ]
|
| 11 |
+
INFO:root:Average loss=0.20628967842994592
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2688 | LCC = 0.8577 | SRCC = 0.8596 ]
|
| 14 |
+
INFO:root:Average loss=0.18531899908796334
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2608 | LCC = 0.8651 | SRCC = 0.8653 ]
|
| 17 |
+
INFO:root:Average loss=0.17028685458577597
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2490 | LCC = 0.8692 | SRCC = 0.8693 ]
|
| 20 |
+
INFO:root:Average loss=0.15651398706130493
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2452 | LCC = 0.8712 | SRCC = 0.8696 ]
|
| 23 |
+
INFO:root:Average loss=0.14767660764165413
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2434 | LCC = 0.8708 | SRCC = 0.8693 ]
|
| 26 |
+
INFO:root:Average loss=0.1454870343590394
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2332 | LCC = 0.8654 | SRCC = 0.8666 ]
|
| 29 |
+
INFO:root:Average loss=0.13086320966100082
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2651 | LCC = 0.8646 | SRCC = 0.8659 ]
|
| 32 |
+
INFO:root:Average loss=0.13209669798230514
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2353 | LCC = 0.8653 | SRCC = 0.8648 ]
|
| 35 |
+
INFO:root:Average loss=0.11594731981555621
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2512 | LCC = 0.8638 | SRCC = 0.8637 ]
|
| 38 |
+
INFO:root:Average loss=0.10751021524461415
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2630 | LCC = 0.8627 | SRCC = 0.8650 ]
|
| 41 |
+
INFO:root:Average loss=0.10847543781766525
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2342 | LCC = 0.8638 | SRCC = 0.8646 ]
|
| 44 |
+
INFO:root:Average loss=0.09495962396837197
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2558 | LCC = 0.8581 | SRCC = 0.8585 ]
|
| 47 |
+
INFO:root:Average loss=0.09150164330808017
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2514 | LCC = 0.8579 | SRCC = 0.8582 ]
|
| 50 |
+
INFO:root:Average loss=0.08520652950765231
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2599 | LCC = 0.8562 | SRCC = 0.8576 ]
|
| 53 |
+
INFO:root:Average loss=0.07926359218664658
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2289 | LCC = 0.8552 | SRCC = 0.8571 ]
|
| 56 |
+
INFO:root:Average loss=0.07587479799985886
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2405 | LCC = 0.8530 | SRCC = 0.8549 ]
|
| 59 |
+
INFO:root:Average loss=0.07495530842779538
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2702 | LCC = 0.8545 | SRCC = 0.8561 ]
|
| 62 |
+
INFO:root:Average loss=0.06786195890834698
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2363 | LCC = 0.8514 | SRCC = 0.8527 ]
|
| 65 |
+
INFO:root:Average loss=0.06390005674881813
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2880 | LCC = 0.8501 | SRCC = 0.8523 ]
|
| 68 |
+
INFO:root:Average loss=0.06459033708923902
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2764 | LCC = 0.8492 | SRCC = 0.8524 ]
|
| 71 |
+
INFO:root:Average loss=0.0626679030366433
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2336 | LCC = 0.8538 | SRCC = 0.8561 ]
|
| 74 |
+
INFO:root:Average loss=0.057929873609772094
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2522 | LCC = 0.8498 | SRCC = 0.8517 ]
|
| 77 |
+
INFO:root:Average loss=0.054576437609891094
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2479 | LCC = 0.8518 | SRCC = 0.8535 ]
|
| 80 |
+
INFO:root:Average loss=0.05767752265987488
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2480 | LCC = 0.8496 | SRCC = 0.8512 ]
|
| 83 |
+
INFO:root:Average loss=0.05206704005981103
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2765 | LCC = 0.8493 | SRCC = 0.8518 ]
|
| 86 |
+
INFO:root:Average loss=0.04931540763339935
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2568 | LCC = 0.8494 | SRCC = 0.8503 ]
|
| 89 |
+
INFO:root:Average loss=0.04757789340920937
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2551 | LCC = 0.8521 | SRCC = 0.8531 ]
|
| 92 |
+
INFO:root:Average loss=0.04776382002119835
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2409 | LCC = 0.8512 | SRCC = 0.8522 ]
|
| 95 |
+
INFO:root:Average loss=0.045480646479588285
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2519 | LCC = 0.8506 | SRCC = 0.8530 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2367 | LCC = 0.8643 | SRCC = 0.8639 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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_8epoch/layer6/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4da41d337bab409df9120bda5f858c3242ee8a66301e702c145e23f333a0aa28
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:56b570322548468a0e5d392aae1ef40f04c7c330bd8bbc864435d326f85340d6
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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.3797180706109756
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3046 | LCC = 0.8346 | SRCC = 0.8365 ]
|
| 8 |
+
INFO:root:Average loss=0.2305016590234561
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2970 | LCC = 0.8560 | SRCC = 0.8567 ]
|
| 11 |
+
INFO:root:Average loss=0.21048951024810472
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2487 | LCC = 0.8620 | SRCC = 0.8611 ]
|
| 14 |
+
INFO:root:Average loss=0.1981869487044139
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2646 | LCC = 0.8666 | SRCC = 0.8650 ]
|
| 17 |
+
INFO:root:Average loss=0.18076091221509835
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2333 | LCC = 0.8649 | SRCC = 0.8658 ]
|
| 20 |
+
INFO:root:Average loss=0.16797982777158418
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2682 | LCC = 0.8642 | SRCC = 0.8651 ]
|
| 23 |
+
INFO:root:Average loss=0.1563210419546335
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2478 | LCC = 0.8661 | SRCC = 0.8650 ]
|
| 26 |
+
INFO:root:Average loss=0.1436955794118918
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2476 | LCC = 0.8625 | SRCC = 0.8622 ]
|
| 29 |
+
INFO:root:Average loss=0.1337132291534008
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2725 | LCC = 0.8544 | SRCC = 0.8546 ]
|
| 32 |
+
INFO:root:Average loss=0.13001818515551397
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2334 | LCC = 0.8595 | SRCC = 0.8588 ]
|
| 35 |
+
INFO:root:Average loss=0.11326501470727798
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2451 | LCC = 0.8579 | SRCC = 0.8586 ]
|
| 38 |
+
INFO:root:Average loss=0.10822764717233486
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2424 | LCC = 0.8543 | SRCC = 0.8550 ]
|
| 41 |
+
INFO:root:Average loss=0.10454240384010169
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2552 | LCC = 0.8536 | SRCC = 0.8525 ]
|
| 44 |
+
INFO:root:Average loss=0.09520155578278579
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2694 | LCC = 0.8526 | SRCC = 0.8515 ]
|
| 47 |
+
INFO:root:Average loss=0.08673519899065678
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2543 | LCC = 0.8538 | SRCC = 0.8536 ]
|
| 50 |
+
INFO:root:Average loss=0.08294013228553992
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2458 | LCC = 0.8495 | SRCC = 0.8504 ]
|
| 53 |
+
INFO:root:Average loss=0.07720066425509942
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2380 | LCC = 0.8510 | SRCC = 0.8518 ]
|
| 56 |
+
INFO:root:Average loss=0.07973640894469543
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2383 | LCC = 0.8531 | SRCC = 0.8532 ]
|
| 59 |
+
INFO:root:Average loss=0.07244081924167964
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2476 | LCC = 0.8460 | SRCC = 0.8452 ]
|
| 62 |
+
INFO:root:Average loss=0.06955408964019555
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2450 | LCC = 0.8496 | SRCC = 0.8494 ]
|
| 65 |
+
INFO:root:Average loss=0.06167036619706032
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2696 | LCC = 0.8440 | SRCC = 0.8436 ]
|
| 68 |
+
INFO:root:Average loss=0.06148498204465096
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2755 | LCC = 0.8439 | SRCC = 0.8430 ]
|
| 71 |
+
INFO:root:Average loss=0.06650885165883945
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2547 | LCC = 0.8477 | SRCC = 0.8471 ]
|
| 74 |
+
INFO:root:Average loss=0.057192399859046325
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2527 | LCC = 0.8450 | SRCC = 0.8445 ]
|
| 77 |
+
INFO:root:Average loss=0.05703940180440744
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2541 | LCC = 0.8491 | SRCC = 0.8501 ]
|
| 80 |
+
INFO:root:Average loss=0.05327783687374531
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2460 | LCC = 0.8481 | SRCC = 0.8479 ]
|
| 83 |
+
INFO:root:Average loss=0.049764264231690995
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2582 | LCC = 0.8453 | SRCC = 0.8447 ]
|
| 86 |
+
INFO:root:Average loss=0.047981671296442166
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2657 | LCC = 0.8437 | SRCC = 0.8419 ]
|
| 89 |
+
INFO:root:Average loss=0.04447451112075494
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2835 | LCC = 0.8452 | SRCC = 0.8450 ]
|
| 92 |
+
INFO:root:Average loss=0.04357468714125645
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2536 | LCC = 0.8481 | SRCC = 0.8473 ]
|
| 95 |
+
INFO:root:Average loss=0.0441230527865581
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2505 | LCC = 0.8434 | SRCC = 0.8419 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2562 | LCC = 0.8669 | SRCC = 0.8678 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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_8epoch/layer6/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c3a340a3c9b00ee154b8cc01d4c70a8607131a1149c8cc7465131a4d8f472467
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e949f710450eb03c290163913d30d804fd3d0b850d3fd878d4e4f335753ea46f
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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.35811989085796553
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2855 | LCC = 0.8356 | SRCC = 0.8394 ]
|
| 8 |
+
INFO:root:Average loss=0.23174308660702828
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2696 | LCC = 0.8587 | SRCC = 0.8571 ]
|
| 11 |
+
INFO:root:Average loss=0.20694455829186317
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2411 | LCC = 0.8652 | SRCC = 0.8645 ]
|
| 14 |
+
INFO:root:Average loss=0.19412387449007767
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2507 | LCC = 0.8690 | SRCC = 0.8672 ]
|
| 17 |
+
INFO:root:Average loss=0.17476236848877028
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2181 | LCC = 0.8662 | SRCC = 0.8685 ]
|
| 20 |
+
INFO:root:Average loss=0.16277041458166563
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2422 | LCC = 0.8689 | SRCC = 0.8669 ]
|
| 23 |
+
INFO:root:Average loss=0.14799939946104318
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2368 | LCC = 0.8692 | SRCC = 0.8682 ]
|
| 26 |
+
INFO:root:Average loss=0.14131302386522293
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2461 | LCC = 0.8671 | SRCC = 0.8668 ]
|
| 29 |
+
INFO:root:Average loss=0.12920137915091637
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2390 | LCC = 0.8635 | SRCC = 0.8640 ]
|
| 32 |
+
INFO:root:Average loss=0.12187185606513268
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2271 | LCC = 0.8669 | SRCC = 0.8675 ]
|
| 35 |
+
INFO:root:Average loss=0.1094062059926681
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2259 | LCC = 0.8613 | SRCC = 0.8616 ]
|
| 38 |
+
INFO:root:Average loss=0.10354185056609985
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2506 | LCC = 0.8603 | SRCC = 0.8608 ]
|
| 41 |
+
INFO:root:Average loss=0.09743791331465428
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2289 | LCC = 0.8613 | SRCC = 0.8616 ]
|
| 44 |
+
INFO:root:Average loss=0.09324750246909949
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2974 | LCC = 0.8539 | SRCC = 0.8557 ]
|
| 47 |
+
INFO:root:Average loss=0.08660409962519622
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2674 | LCC = 0.8559 | SRCC = 0.8572 ]
|
| 50 |
+
INFO:root:Average loss=0.0802232102992443
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2495 | LCC = 0.8566 | SRCC = 0.8564 ]
|
| 53 |
+
INFO:root:Average loss=0.07793291405034371
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2467 | LCC = 0.8566 | SRCC = 0.8556 ]
|
| 56 |
+
INFO:root:Average loss=0.07416404583133183
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2503 | LCC = 0.8575 | SRCC = 0.8589 ]
|
| 59 |
+
INFO:root:Average loss=0.07246936599795635
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2352 | LCC = 0.8545 | SRCC = 0.8548 ]
|
| 62 |
+
INFO:root:Average loss=0.06706997064443734
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2268 | LCC = 0.8561 | SRCC = 0.8555 ]
|
| 65 |
+
INFO:root:Average loss=0.06369093447350539
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2389 | LCC = 0.8555 | SRCC = 0.8549 ]
|
| 68 |
+
INFO:root:Average loss=0.057506588072730944
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2496 | LCC = 0.8510 | SRCC = 0.8510 ]
|
| 71 |
+
INFO:root:Average loss=0.06347753059787628
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2309 | LCC = 0.8569 | SRCC = 0.8567 ]
|
| 74 |
+
INFO:root:Average loss=0.055077012532796614
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2539 | LCC = 0.8537 | SRCC = 0.8541 ]
|
| 77 |
+
INFO:root:Average loss=0.05302810387160534
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2626 | LCC = 0.8524 | SRCC = 0.8529 ]
|
| 80 |
+
INFO:root:Average loss=0.05048807672201059
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2428 | LCC = 0.8513 | SRCC = 0.8510 ]
|
| 83 |
+
INFO:root:Average loss=0.05132855040331682
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2463 | LCC = 0.8525 | SRCC = 0.8521 ]
|
| 86 |
+
INFO:root:Average loss=0.046166109470411755
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2623 | LCC = 0.8507 | SRCC = 0.8507 ]
|
| 89 |
+
INFO:root:Average loss=0.04295750248890657
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2365 | LCC = 0.8521 | SRCC = 0.8527 ]
|
| 92 |
+
INFO:root:Average loss=0.04421266321188364
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2431 | LCC = 0.8536 | SRCC = 0.8542 ]
|
| 95 |
+
INFO:root:Average loss=0.04202679036041865
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2473 | LCC = 0.8542 | SRCC = 0.8547 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2523 | LCC = 0.8611 | SRCC = 0.8646 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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_8epoch/layer6/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:90a5e9750e159716c95c09fb2e3ad10009d90142d3ecc0dc5c7137a955376e50
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b4d858653d04b8c64d1c8ce6ece083d1c3e874c6c955e93f2d0707ce68f7482e
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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.3604837806943135
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3241 | LCC = 0.8343 | SRCC = 0.8350 ]
|
| 8 |
+
INFO:root:Average loss=0.2331581534101413
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2768 | LCC = 0.8506 | SRCC = 0.8496 ]
|
| 11 |
+
INFO:root:Average loss=0.2057274090938079
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2501 | LCC = 0.8595 | SRCC = 0.8580 ]
|
| 14 |
+
INFO:root:Average loss=0.191218614196166
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2560 | LCC = 0.8631 | SRCC = 0.8616 ]
|
| 17 |
+
INFO:root:Average loss=0.17719015383567566
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2326 | LCC = 0.8634 | SRCC = 0.8630 ]
|
| 20 |
+
INFO:root:Average loss=0.17022792326334196
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2758 | LCC = 0.8614 | SRCC = 0.8604 ]
|
| 23 |
+
INFO:root:Average loss=0.1579711240453598
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2327 | LCC = 0.8606 | SRCC = 0.8610 ]
|
| 26 |
+
INFO:root:Average loss=0.14690540969753876
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2559 | LCC = 0.8593 | SRCC = 0.8603 ]
|
| 29 |
+
INFO:root:Average loss=0.13715836730522987
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2581 | LCC = 0.8587 | SRCC = 0.8578 ]
|
| 32 |
+
INFO:root:Average loss=0.12865965249828804
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2611 | LCC = 0.8561 | SRCC = 0.8558 ]
|
| 35 |
+
INFO:root:Average loss=0.11679353803778306
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2572 | LCC = 0.8575 | SRCC = 0.8581 ]
|
| 38 |
+
INFO:root:Average loss=0.1120085115902699
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2507 | LCC = 0.8551 | SRCC = 0.8558 ]
|
| 41 |
+
INFO:root:Average loss=0.10751390299544884
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2619 | LCC = 0.8516 | SRCC = 0.8521 ]
|
| 44 |
+
INFO:root:Average loss=0.10514366779571925
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2485 | LCC = 0.8506 | SRCC = 0.8511 ]
|
| 47 |
+
INFO:root:Average loss=0.09369600846026188
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2562 | LCC = 0.8512 | SRCC = 0.8518 ]
|
| 50 |
+
INFO:root:Average loss=0.08671288693753573
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2629 | LCC = 0.8497 | SRCC = 0.8518 ]
|
| 53 |
+
INFO:root:Average loss=0.08395195432389393
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2715 | LCC = 0.8498 | SRCC = 0.8515 ]
|
| 56 |
+
INFO:root:Average loss=0.08351552882828774
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2751 | LCC = 0.8487 | SRCC = 0.8502 ]
|
| 59 |
+
INFO:root:Average loss=0.08010196704895069
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2838 | LCC = 0.8459 | SRCC = 0.8481 ]
|
| 62 |
+
INFO:root:Average loss=0.0744340203415889
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2526 | LCC = 0.8501 | SRCC = 0.8512 ]
|
| 65 |
+
INFO:root:Average loss=0.06690540408285764
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2752 | LCC = 0.8450 | SRCC = 0.8464 ]
|
| 68 |
+
INFO:root:Average loss=0.07163178252104001
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2750 | LCC = 0.8497 | SRCC = 0.8523 ]
|
| 71 |
+
INFO:root:Average loss=0.06305020106717563
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2566 | LCC = 0.8493 | SRCC = 0.8493 ]
|
| 74 |
+
INFO:root:Average loss=0.05843058061332275
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2525 | LCC = 0.8448 | SRCC = 0.8452 ]
|
| 77 |
+
INFO:root:Average loss=0.05796974214414755
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2625 | LCC = 0.8451 | SRCC = 0.8460 ]
|
| 80 |
+
INFO:root:Average loss=0.05792016645845694
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2687 | LCC = 0.8429 | SRCC = 0.8440 ]
|
| 83 |
+
INFO:root:Average loss=0.0553492755175401
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2491 | LCC = 0.8500 | SRCC = 0.8518 ]
|
| 86 |
+
INFO:root:Average loss=0.05292717873668059
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2502 | LCC = 0.8471 | SRCC = 0.8483 ]
|
| 89 |
+
INFO:root:Average loss=0.04914245057182434
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2660 | LCC = 0.8468 | SRCC = 0.8488 ]
|
| 92 |
+
INFO:root:Average loss=0.046933587974844836
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2531 | LCC = 0.8474 | SRCC = 0.8484 ]
|
| 95 |
+
INFO:root:Average loss=0.04748835655836723
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2588 | LCC = 0.8448 | SRCC = 0.8453 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2281 | LCC = 0.8657 | SRCC = 0.8675 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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_8epoch/layer6/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:397f58628d505f682ce8eef89e81c31956362c4e51ff7457d979973ae6441cfe
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5aed4c0e3a4ffc26b47e238b8ba399c924795cf8d7087008e04d50b87280e34e
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer6/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.3770978162303949
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3275 | LCC = 0.8323 | SRCC = 0.8332 ]
|
| 8 |
+
INFO:root:Average loss=0.2387025990547278
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2596 | LCC = 0.8497 | SRCC = 0.8498 ]
|
| 11 |
+
INFO:root:Average loss=0.21004770352290228
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2532 | LCC = 0.8611 | SRCC = 0.8606 ]
|
| 14 |
+
INFO:root:Average loss=0.1903307160887963
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2562 | LCC = 0.8654 | SRCC = 0.8620 ]
|
| 17 |
+
INFO:root:Average loss=0.17257102416493955
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2605 | LCC = 0.8643 | SRCC = 0.8634 ]
|
| 20 |
+
INFO:root:Average loss=0.16535658886035284
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2388 | LCC = 0.8655 | SRCC = 0.8653 ]
|
| 23 |
+
INFO:root:Average loss=0.15282277906170258
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2265 | LCC = 0.8653 | SRCC = 0.8646 ]
|
| 26 |
+
INFO:root:Average loss=0.1427481387479183
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2672 | LCC = 0.8649 | SRCC = 0.8616 ]
|
| 29 |
+
INFO:root:Average loss=0.13475456872047523
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2567 | LCC = 0.8611 | SRCC = 0.8585 ]
|
| 32 |
+
INFO:root:Average loss=0.12519202104363686
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2382 | LCC = 0.8615 | SRCC = 0.8595 ]
|
| 35 |
+
INFO:root:Average loss=0.11535744235301629
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2624 | LCC = 0.8616 | SRCC = 0.8614 ]
|
| 38 |
+
INFO:root:Average loss=0.10666486000021298
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2487 | LCC = 0.8593 | SRCC = 0.8570 ]
|
| 41 |
+
INFO:root:Average loss=0.1015381939136065
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2378 | LCC = 0.8577 | SRCC = 0.8554 ]
|
| 44 |
+
INFO:root:Average loss=0.0960148461163044
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2706 | LCC = 0.8567 | SRCC = 0.8539 ]
|
| 47 |
+
INFO:root:Average loss=0.09119731350204884
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2399 | LCC = 0.8560 | SRCC = 0.8547 ]
|
| 50 |
+
INFO:root:Average loss=0.081835809808511
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2524 | LCC = 0.8541 | SRCC = 0.8513 ]
|
| 53 |
+
INFO:root:Average loss=0.08057321904179378
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2529 | LCC = 0.8497 | SRCC = 0.8479 ]
|
| 56 |
+
INFO:root:Average loss=0.07568612451163623
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2479 | LCC = 0.8510 | SRCC = 0.8491 ]
|
| 59 |
+
INFO:root:Average loss=0.070625869748302
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2463 | LCC = 0.8553 | SRCC = 0.8540 ]
|
| 62 |
+
INFO:root:Average loss=0.06824355677534373
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2423 | LCC = 0.8536 | SRCC = 0.8517 ]
|
| 65 |
+
INFO:root:Average loss=0.06662077552232987
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2452 | LCC = 0.8556 | SRCC = 0.8540 ]
|
| 68 |
+
INFO:root:Average loss=0.059250764978619724
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2390 | LCC = 0.8555 | SRCC = 0.8540 ]
|
| 71 |
+
INFO:root:Average loss=0.058074076349536576
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2421 | LCC = 0.8544 | SRCC = 0.8522 ]
|
| 74 |
+
INFO:root:Average loss=0.05762129979064831
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2617 | LCC = 0.8510 | SRCC = 0.8492 ]
|
| 77 |
+
INFO:root:Average loss=0.056809222230162375
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2373 | LCC = 0.8537 | SRCC = 0.8508 ]
|
| 80 |
+
INFO:root:Average loss=0.05110459559811996
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2363 | LCC = 0.8545 | SRCC = 0.8540 ]
|
| 83 |
+
INFO:root:Average loss=0.05058353465910141
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2560 | LCC = 0.8498 | SRCC = 0.8479 ]
|
| 86 |
+
INFO:root:Average loss=0.04859237482723517
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2396 | LCC = 0.8527 | SRCC = 0.8519 ]
|
| 89 |
+
INFO:root:Average loss=0.04709915105157938
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2621 | LCC = 0.8506 | SRCC = 0.8497 ]
|
| 92 |
+
INFO:root:Average loss=0.04966300873993299
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2490 | LCC = 0.8528 | SRCC = 0.8523 ]
|
| 95 |
+
INFO:root:Average loss=0.04379731996987875
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2431 | LCC = 0.8532 | SRCC = 0.8525 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2391 | LCC = 0.8677 | SRCC = 0.8670 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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_8epoch/layer7/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bcb23afd3177b4134fd8992391887bb35558ec5236e983a5ea6ef140639c1514
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3d95505fe3bca9aafb0219284b64b790e45615f1a520ecad914672b76849e621
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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.3469079965964342
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2989 | LCC = 0.8382 | SRCC = 0.8401 ]
|
| 8 |
+
INFO:root:Average loss=0.23198478783552462
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2585 | LCC = 0.8558 | SRCC = 0.8529 ]
|
| 11 |
+
INFO:root:Average loss=0.20541685781417748
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2483 | LCC = 0.8641 | SRCC = 0.8621 ]
|
| 14 |
+
INFO:root:Average loss=0.19130761921405792
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2713 | LCC = 0.8677 | SRCC = 0.8651 ]
|
| 17 |
+
INFO:root:Average loss=0.18461942663177466
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2428 | LCC = 0.8685 | SRCC = 0.8679 ]
|
| 20 |
+
INFO:root:Average loss=0.16500769097071427
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2516 | LCC = 0.8690 | SRCC = 0.8660 ]
|
| 23 |
+
INFO:root:Average loss=0.15804373329648605
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2455 | LCC = 0.8693 | SRCC = 0.8679 ]
|
| 26 |
+
INFO:root:Average loss=0.15088937603510344
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2357 | LCC = 0.8691 | SRCC = 0.8673 ]
|
| 29 |
+
INFO:root:Average loss=0.1387404889250413
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2327 | LCC = 0.8666 | SRCC = 0.8658 ]
|
| 32 |
+
INFO:root:Average loss=0.13043687473504972
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2570 | LCC = 0.8636 | SRCC = 0.8635 ]
|
| 35 |
+
INFO:root:Average loss=0.11843642649742273
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2293 | LCC = 0.8645 | SRCC = 0.8642 ]
|
| 38 |
+
INFO:root:Average loss=0.12040004864908181
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2339 | LCC = 0.8614 | SRCC = 0.8606 ]
|
| 41 |
+
INFO:root:Average loss=0.11031159605735387
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2628 | LCC = 0.8545 | SRCC = 0.8545 ]
|
| 44 |
+
INFO:root:Average loss=0.10428104143685256
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2402 | LCC = 0.8574 | SRCC = 0.8572 ]
|
| 47 |
+
INFO:root:Average loss=0.09424971884641892
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2438 | LCC = 0.8565 | SRCC = 0.8566 ]
|
| 50 |
+
INFO:root:Average loss=0.08903348732453126
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2415 | LCC = 0.8514 | SRCC = 0.8501 ]
|
| 53 |
+
INFO:root:Average loss=0.08564536870481113
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2446 | LCC = 0.8547 | SRCC = 0.8534 ]
|
| 56 |
+
INFO:root:Average loss=0.08497493747526254
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2353 | LCC = 0.8549 | SRCC = 0.8540 ]
|
| 59 |
+
INFO:root:Average loss=0.07745299349801663
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2783 | LCC = 0.8477 | SRCC = 0.8468 ]
|
| 62 |
+
INFO:root:Average loss=0.07135221438530163
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2526 | LCC = 0.8502 | SRCC = 0.8495 ]
|
| 65 |
+
INFO:root:Average loss=0.07131569780027255
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2605 | LCC = 0.8463 | SRCC = 0.8451 ]
|
| 68 |
+
INFO:root:Average loss=0.06440688287600493
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2764 | LCC = 0.8484 | SRCC = 0.8474 ]
|
| 71 |
+
INFO:root:Average loss=0.0627313334112748
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2873 | LCC = 0.8452 | SRCC = 0.8444 ]
|
| 74 |
+
INFO:root:Average loss=0.05825639129258119
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2651 | LCC = 0.8423 | SRCC = 0.8421 ]
|
| 77 |
+
INFO:root:Average loss=0.056449636578177795
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2629 | LCC = 0.8450 | SRCC = 0.8436 ]
|
| 80 |
+
INFO:root:Average loss=0.055312554948987104
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2510 | LCC = 0.8484 | SRCC = 0.8479 ]
|
| 83 |
+
INFO:root:Average loss=0.051834007199758135
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2352 | LCC = 0.8502 | SRCC = 0.8500 ]
|
| 86 |
+
INFO:root:Average loss=0.04881216726528528
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2513 | LCC = 0.8447 | SRCC = 0.8440 ]
|
| 89 |
+
INFO:root:Average loss=0.04795687951338597
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2557 | LCC = 0.8469 | SRCC = 0.8456 ]
|
| 92 |
+
INFO:root:Average loss=0.045337436434168085
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2537 | LCC = 0.8483 | SRCC = 0.8473 ]
|
| 95 |
+
INFO:root:Average loss=0.04486588393457425
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2551 | LCC = 0.8463 | SRCC = 0.8444 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2494 | LCC = 0.8673 | SRCC = 0.8665 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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_8epoch/layer7/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:53753beeb216822b7a44963f08a62439e29026986a49876121c9e28db9c01ac6
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:22e32b5b8e6cff4723c3767c557a1cf8849e21f40cde73a0161d2cbd38cc58cc
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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.35092105181553424
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3467 | LCC = 0.8330 | SRCC = 0.8389 ]
|
| 8 |
+
INFO:root:Average loss=0.22391803409808722
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2580 | LCC = 0.8565 | SRCC = 0.8579 ]
|
| 11 |
+
INFO:root:Average loss=0.2077186908095311
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2649 | LCC = 0.8597 | SRCC = 0.8611 ]
|
| 14 |
+
INFO:root:Average loss=0.1873732989606185
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2662 | LCC = 0.8629 | SRCC = 0.8674 ]
|
| 17 |
+
INFO:root:Average loss=0.1828128002010859
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2472 | LCC = 0.8694 | SRCC = 0.8691 ]
|
| 20 |
+
INFO:root:Average loss=0.17032867536330834
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2253 | LCC = 0.8675 | SRCC = 0.8682 ]
|
| 23 |
+
INFO:root:Average loss=0.15470075215666723
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2232 | LCC = 0.8677 | SRCC = 0.8674 ]
|
| 26 |
+
INFO:root:Average loss=0.14759624739870048
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2272 | LCC = 0.8673 | SRCC = 0.8694 ]
|
| 29 |
+
INFO:root:Average loss=0.14010418645846537
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2238 | LCC = 0.8671 | SRCC = 0.8693 ]
|
| 32 |
+
INFO:root:Average loss=0.1272563649675785
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2293 | LCC = 0.8661 | SRCC = 0.8664 ]
|
| 35 |
+
INFO:root:Average loss=0.12291097459502709
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2458 | LCC = 0.8646 | SRCC = 0.8627 ]
|
| 38 |
+
INFO:root:Average loss=0.11045287401439288
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2423 | LCC = 0.8621 | SRCC = 0.8611 ]
|
| 41 |
+
INFO:root:Average loss=0.10909775281563783
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2404 | LCC = 0.8589 | SRCC = 0.8595 ]
|
| 44 |
+
INFO:root:Average loss=0.09805169773216431
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2434 | LCC = 0.8584 | SRCC = 0.8595 ]
|
| 47 |
+
INFO:root:Average loss=0.09028928206326106
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2443 | LCC = 0.8571 | SRCC = 0.8571 ]
|
| 50 |
+
INFO:root:Average loss=0.08622201809134239
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2455 | LCC = 0.8500 | SRCC = 0.8495 ]
|
| 53 |
+
INFO:root:Average loss=0.08073516118411835
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2566 | LCC = 0.8536 | SRCC = 0.8546 ]
|
| 56 |
+
INFO:root:Average loss=0.07807559548662259
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2554 | LCC = 0.8526 | SRCC = 0.8529 ]
|
| 59 |
+
INFO:root:Average loss=0.07593226824433376
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2307 | LCC = 0.8567 | SRCC = 0.8576 ]
|
| 62 |
+
INFO:root:Average loss=0.07104522601152077
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2343 | LCC = 0.8569 | SRCC = 0.8568 ]
|
| 65 |
+
INFO:root:Average loss=0.0637034338254195
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2489 | LCC = 0.8500 | SRCC = 0.8502 ]
|
| 68 |
+
INFO:root:Average loss=0.06301592171001129
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2420 | LCC = 0.8489 | SRCC = 0.8498 ]
|
| 71 |
+
INFO:root:Average loss=0.05648809709610083
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2530 | LCC = 0.8474 | SRCC = 0.8473 ]
|
| 74 |
+
INFO:root:Average loss=0.0569760552010475
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2716 | LCC = 0.8537 | SRCC = 0.8538 ]
|
| 77 |
+
INFO:root:Average loss=0.054729500642189614
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2478 | LCC = 0.8465 | SRCC = 0.8482 ]
|
| 80 |
+
INFO:root:Average loss=0.05050178380826345
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2516 | LCC = 0.8517 | SRCC = 0.8526 ]
|
| 83 |
+
INFO:root:Average loss=0.0497330358872811
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2371 | LCC = 0.8499 | SRCC = 0.8501 ]
|
| 86 |
+
INFO:root:Average loss=0.04755664182205995
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2439 | LCC = 0.8511 | SRCC = 0.8500 ]
|
| 89 |
+
INFO:root:Average loss=0.046851443532758795
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2498 | LCC = 0.8515 | SRCC = 0.8519 ]
|
| 92 |
+
INFO:root:Average loss=0.04367782999403202
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2481 | LCC = 0.8487 | SRCC = 0.8492 ]
|
| 95 |
+
INFO:root:Average loss=0.04227986633300017
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2561 | LCC = 0.8521 | SRCC = 0.8512 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2591 | LCC = 0.8659 | SRCC = 0.8697 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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_8epoch/layer7/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:808f410352b17bde41dd47121471350a76295d454545e44e38a25e69823b3dff
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e50b120f4936b41a4e400e3d81050ab5235850aa90ab0561a6fd79bd6b6b76f9
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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.3361775741363183
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3013 | LCC = 0.8406 | SRCC = 0.8431 ]
|
| 8 |
+
INFO:root:Average loss=0.22454409377697188
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3010 | LCC = 0.8625 | SRCC = 0.8594 ]
|
| 11 |
+
INFO:root:Average loss=0.20117125765253335
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2633 | LCC = 0.8674 | SRCC = 0.8654 ]
|
| 14 |
+
INFO:root:Average loss=0.18968014877576095
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2349 | LCC = 0.8672 | SRCC = 0.8676 ]
|
| 17 |
+
INFO:root:Average loss=0.17348546171799684
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2443 | LCC = 0.8684 | SRCC = 0.8675 ]
|
| 20 |
+
INFO:root:Average loss=0.1605497519366252
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2410 | LCC = 0.8680 | SRCC = 0.8677 ]
|
| 23 |
+
INFO:root:Average loss=0.16413550499158028
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2729 | LCC = 0.8683 | SRCC = 0.8689 ]
|
| 26 |
+
INFO:root:Average loss=0.1432249536498999
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2575 | LCC = 0.8671 | SRCC = 0.8662 ]
|
| 29 |
+
INFO:root:Average loss=0.13707019159427056
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2333 | LCC = 0.8670 | SRCC = 0.8674 ]
|
| 32 |
+
INFO:root:Average loss=0.12744018387718078
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2412 | LCC = 0.8634 | SRCC = 0.8633 ]
|
| 35 |
+
INFO:root:Average loss=0.11622359966620421
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2643 | LCC = 0.8639 | SRCC = 0.8639 ]
|
| 38 |
+
INFO:root:Average loss=0.11292111424681468
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2803 | LCC = 0.8580 | SRCC = 0.8582 ]
|
| 41 |
+
INFO:root:Average loss=0.10526881997401898
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2425 | LCC = 0.8621 | SRCC = 0.8636 ]
|
| 44 |
+
INFO:root:Average loss=0.10095537816866851
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2605 | LCC = 0.8557 | SRCC = 0.8590 ]
|
| 47 |
+
INFO:root:Average loss=0.09372408124498832
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2570 | LCC = 0.8530 | SRCC = 0.8542 ]
|
| 50 |
+
INFO:root:Average loss=0.08658124324985039
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2685 | LCC = 0.8596 | SRCC = 0.8607 ]
|
| 53 |
+
INFO:root:Average loss=0.0811407636755552
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2395 | LCC = 0.8570 | SRCC = 0.8582 ]
|
| 56 |
+
INFO:root:Average loss=0.07503594157214348
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2475 | LCC = 0.8559 | SRCC = 0.8550 ]
|
| 59 |
+
INFO:root:Average loss=0.07276743564468163
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2805 | LCC = 0.8557 | SRCC = 0.8550 ]
|
| 62 |
+
INFO:root:Average loss=0.07363468394256555
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2336 | LCC = 0.8569 | SRCC = 0.8587 ]
|
| 65 |
+
INFO:root:Average loss=0.07126202931006749
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2502 | LCC = 0.8539 | SRCC = 0.8541 ]
|
| 68 |
+
INFO:root:Average loss=0.0614911320213324
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2507 | LCC = 0.8570 | SRCC = 0.8577 ]
|
| 71 |
+
INFO:root:Average loss=0.06123488305661923
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2405 | LCC = 0.8536 | SRCC = 0.8550 ]
|
| 74 |
+
INFO:root:Average loss=0.05606937422775305
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2599 | LCC = 0.8507 | SRCC = 0.8510 ]
|
| 77 |
+
INFO:root:Average loss=0.05564926383204949
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2626 | LCC = 0.8530 | SRCC = 0.8536 ]
|
| 80 |
+
INFO:root:Average loss=0.05060609640219273
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2654 | LCC = 0.8502 | SRCC = 0.8512 ]
|
| 83 |
+
INFO:root:Average loss=0.05456724017858505
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2618 | LCC = 0.8515 | SRCC = 0.8526 ]
|
| 86 |
+
INFO:root:Average loss=0.05297986059807814
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2483 | LCC = 0.8501 | SRCC = 0.8511 ]
|
| 89 |
+
INFO:root:Average loss=0.04690793209159986
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2490 | LCC = 0.8527 | SRCC = 0.8530 ]
|
| 92 |
+
INFO:root:Average loss=0.04672756762458728
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2377 | LCC = 0.8532 | SRCC = 0.8547 ]
|
| 95 |
+
INFO:root:Average loss=0.051660896016237065
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2572 | LCC = 0.8520 | SRCC = 0.8522 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2382 | LCC = 0.8607 | SRCC = 0.8628 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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_8epoch/layer7/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:49619c4ec3b71f6a7b998d5fb817bd1de77e54dd5338a2cba438d42e85a6f417
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9f6d6aff373ff50a73cacdb16fce4025d6a77eaf15c43ae9bd6556f1aa0d0cc2
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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.36031487526801914
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2909 | LCC = 0.8361 | SRCC = 0.8392 ]
|
| 8 |
+
INFO:root:Average loss=0.22986766352103308
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2835 | LCC = 0.8539 | SRCC = 0.8506 ]
|
| 11 |
+
INFO:root:Average loss=0.21281702033220193
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2674 | LCC = 0.8612 | SRCC = 0.8583 ]
|
| 14 |
+
INFO:root:Average loss=0.19140471250582963
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2410 | LCC = 0.8623 | SRCC = 0.8600 ]
|
| 17 |
+
INFO:root:Average loss=0.18375769677834633
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2631 | LCC = 0.8599 | SRCC = 0.8586 ]
|
| 20 |
+
INFO:root:Average loss=0.17224495705121604
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2364 | LCC = 0.8620 | SRCC = 0.8609 ]
|
| 23 |
+
INFO:root:Average loss=0.15913706100904024
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2423 | LCC = 0.8638 | SRCC = 0.8621 ]
|
| 26 |
+
INFO:root:Average loss=0.155670099820082
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2358 | LCC = 0.8633 | SRCC = 0.8627 ]
|
| 29 |
+
INFO:root:Average loss=0.14069017166128525
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2496 | LCC = 0.8571 | SRCC = 0.8553 ]
|
| 32 |
+
INFO:root:Average loss=0.1286107660868229
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2474 | LCC = 0.8587 | SRCC = 0.8572 ]
|
| 35 |
+
INFO:root:Average loss=0.12176207672709073
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2495 | LCC = 0.8587 | SRCC = 0.8576 ]
|
| 38 |
+
INFO:root:Average loss=0.11548341027437112
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2463 | LCC = 0.8580 | SRCC = 0.8565 ]
|
| 41 |
+
INFO:root:Average loss=0.10905310291892444
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2554 | LCC = 0.8566 | SRCC = 0.8552 ]
|
| 44 |
+
INFO:root:Average loss=0.10277998423538147
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2511 | LCC = 0.8490 | SRCC = 0.8482 ]
|
| 47 |
+
INFO:root:Average loss=0.09820752328213973
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2500 | LCC = 0.8500 | SRCC = 0.8498 ]
|
| 50 |
+
INFO:root:Average loss=0.09047077357386932
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2468 | LCC = 0.8519 | SRCC = 0.8514 ]
|
| 53 |
+
INFO:root:Average loss=0.08370262503815003
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2386 | LCC = 0.8501 | SRCC = 0.8502 ]
|
| 56 |
+
INFO:root:Average loss=0.0783373729731792
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2434 | LCC = 0.8489 | SRCC = 0.8481 ]
|
| 59 |
+
INFO:root:Average loss=0.07786965699723133
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2333 | LCC = 0.8554 | SRCC = 0.8550 ]
|
| 62 |
+
INFO:root:Average loss=0.07153331612547238
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2397 | LCC = 0.8520 | SRCC = 0.8514 ]
|
| 65 |
+
INFO:root:Average loss=0.06711864270842992
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2385 | LCC = 0.8527 | SRCC = 0.8528 ]
|
| 68 |
+
INFO:root:Average loss=0.06474789690512878
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2565 | LCC = 0.8452 | SRCC = 0.8442 ]
|
| 71 |
+
INFO:root:Average loss=0.06640106401382348
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2527 | LCC = 0.8476 | SRCC = 0.8468 ]
|
| 74 |
+
INFO:root:Average loss=0.0617793684777541
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2416 | LCC = 0.8485 | SRCC = 0.8480 ]
|
| 77 |
+
INFO:root:Average loss=0.05555884659481354
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2528 | LCC = 0.8471 | SRCC = 0.8463 ]
|
| 80 |
+
INFO:root:Average loss=0.05438259654702284
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2425 | LCC = 0.8511 | SRCC = 0.8501 ]
|
| 83 |
+
INFO:root:Average loss=0.050841256307485774
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2370 | LCC = 0.8497 | SRCC = 0.8503 ]
|
| 86 |
+
INFO:root:Average loss=0.05048329660143608
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2482 | LCC = 0.8499 | SRCC = 0.8490 ]
|
| 89 |
+
INFO:root:Average loss=0.047865212871096074
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2448 | LCC = 0.8464 | SRCC = 0.8446 ]
|
| 92 |
+
INFO:root:Average loss=0.04802763521766815
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2605 | LCC = 0.8441 | SRCC = 0.8437 ]
|
| 95 |
+
INFO:root:Average loss=0.0466090687431204
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2672 | LCC = 0.8404 | SRCC = 0.8388 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2572 | LCC = 0.8572 | SRCC = 0.8603 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/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_8epoch/layer7/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b91b223ae33e3bae2145beadebba1e4d53b06115e58c1332019eace278d7cdca
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_8epoch/layer7/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5b64ab0e2aaa9ab1dff702b1235d92f51ae9897c2669acc2fb94579a4617f413
|
| 3 |
+
size 2749702
|