Upload SSL Layer MOS checkpoints (part 5)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/model_best.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/train.log +99 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/config.gin +31 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/model.pt +3 -0
- results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/model_best.pt +3 -0
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ab412a40c6a2a7fb78b6ba80e6a890b9b5dc58db1ebf1b0ea16c266fa97849e5
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7698ab5576e7ac936a82b6fbf0546adbc3155970ce8bf6022014b940076b93e5
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/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.4333679804053062
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3561 | LCC = 0.8064 | SRCC = 0.8111 ]
|
| 8 |
+
INFO:root:Average loss=0.27634325871864956
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3084 | LCC = 0.8394 | SRCC = 0.8386 ]
|
| 11 |
+
INFO:root:Average loss=0.2417834808047001
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3485 | LCC = 0.8483 | SRCC = 0.8507 ]
|
| 14 |
+
INFO:root:Average loss=0.2363869091257071
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2711 | LCC = 0.8551 | SRCC = 0.8538 ]
|
| 17 |
+
INFO:root:Average loss=0.21763800313839546
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2544 | LCC = 0.8584 | SRCC = 0.8584 ]
|
| 20 |
+
INFO:root:Average loss=0.19965359090994567
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3070 | LCC = 0.8598 | SRCC = 0.8588 ]
|
| 23 |
+
INFO:root:Average loss=0.18454301662934133
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2501 | LCC = 0.8572 | SRCC = 0.8555 ]
|
| 26 |
+
INFO:root:Average loss=0.17260222862928343
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2483 | LCC = 0.8597 | SRCC = 0.8598 ]
|
| 29 |
+
INFO:root:Average loss=0.16166870964643282
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2490 | LCC = 0.8598 | SRCC = 0.8600 ]
|
| 32 |
+
INFO:root:Average loss=0.14926960261968467
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2632 | LCC = 0.8607 | SRCC = 0.8602 ]
|
| 35 |
+
INFO:root:Average loss=0.14366154783429244
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2406 | LCC = 0.8575 | SRCC = 0.8566 ]
|
| 38 |
+
INFO:root:Average loss=0.13270336131636912
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2480 | LCC = 0.8566 | SRCC = 0.8561 ]
|
| 41 |
+
INFO:root:Average loss=0.12664856838110167
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2659 | LCC = 0.8546 | SRCC = 0.8570 ]
|
| 44 |
+
INFO:root:Average loss=0.11981937614006874
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2470 | LCC = 0.8519 | SRCC = 0.8529 ]
|
| 47 |
+
INFO:root:Average loss=0.1176408742291805
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2629 | LCC = 0.8542 | SRCC = 0.8551 ]
|
| 50 |
+
INFO:root:Average loss=0.10629363157428227
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2790 | LCC = 0.8513 | SRCC = 0.8532 ]
|
| 53 |
+
INFO:root:Average loss=0.09905539023188445
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2683 | LCC = 0.8479 | SRCC = 0.8480 ]
|
| 56 |
+
INFO:root:Average loss=0.09355268398156533
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2807 | LCC = 0.8432 | SRCC = 0.8459 ]
|
| 59 |
+
INFO:root:Average loss=0.0945625029122218
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2507 | LCC = 0.8487 | SRCC = 0.8497 ]
|
| 62 |
+
INFO:root:Average loss=0.0855995132946051
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2653 | LCC = 0.8466 | SRCC = 0.8466 ]
|
| 65 |
+
INFO:root:Average loss=0.08877791110903789
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2444 | LCC = 0.8511 | SRCC = 0.8523 ]
|
| 68 |
+
INFO:root:Average loss=0.08188509148282883
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2533 | LCC = 0.8484 | SRCC = 0.8497 ]
|
| 71 |
+
INFO:root:Average loss=0.07404621523351242
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2624 | LCC = 0.8462 | SRCC = 0.8473 ]
|
| 74 |
+
INFO:root:Average loss=0.07226133031340745
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2529 | LCC = 0.8506 | SRCC = 0.8515 ]
|
| 77 |
+
INFO:root:Average loss=0.06969021642819428
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2734 | LCC = 0.8433 | SRCC = 0.8429 ]
|
| 80 |
+
INFO:root:Average loss=0.0658117977854533
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2387 | LCC = 0.8498 | SRCC = 0.8512 ]
|
| 83 |
+
INFO:root:Average loss=0.06294814812449309
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2505 | LCC = 0.8490 | SRCC = 0.8490 ]
|
| 86 |
+
INFO:root:Average loss=0.05886615440249443
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2603 | LCC = 0.8470 | SRCC = 0.8474 ]
|
| 89 |
+
INFO:root:Average loss=0.06502897827289043
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2498 | LCC = 0.8478 | SRCC = 0.8487 ]
|
| 92 |
+
INFO:root:Average loss=0.05716708899499514
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2474 | LCC = 0.8491 | SRCC = 0.8506 ]
|
| 95 |
+
INFO:root:Average loss=0.05529519208730795
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2509 | LCC = 0.8480 | SRCC = 0.8496 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2618 | LCC = 0.8545 | SRCC = 0.8545 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/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_2epoch/layer7/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5c044c654a235ea22bca38526a79abc04b31e7b75fec343bedeefdf651b13165
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:799fe32dda6351c6f9545fff54e5d9ffdf44818f21b1e8eb9cb0347b0a8745dc
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/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.43758178807986087
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3539 | LCC = 0.8074 | SRCC = 0.8132 ]
|
| 8 |
+
INFO:root:Average loss=0.27459769141979706
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3049 | LCC = 0.8408 | SRCC = 0.8401 ]
|
| 11 |
+
INFO:root:Average loss=0.2429573260820829
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3024 | LCC = 0.8515 | SRCC = 0.8493 ]
|
| 14 |
+
INFO:root:Average loss=0.22346308005925936
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2774 | LCC = 0.8556 | SRCC = 0.8543 ]
|
| 17 |
+
INFO:root:Average loss=0.21441376257019165
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2575 | LCC = 0.8587 | SRCC = 0.8564 ]
|
| 20 |
+
INFO:root:Average loss=0.20390632767707872
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2398 | LCC = 0.8612 | SRCC = 0.8603 ]
|
| 23 |
+
INFO:root:Average loss=0.1827615274068637
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2447 | LCC = 0.8647 | SRCC = 0.8631 ]
|
| 26 |
+
INFO:root:Average loss=0.17666180852131966
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2448 | LCC = 0.8626 | SRCC = 0.8612 ]
|
| 29 |
+
INFO:root:Average loss=0.17401467999204612
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2554 | LCC = 0.8610 | SRCC = 0.8603 ]
|
| 32 |
+
INFO:root:Average loss=0.15118789195250243
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2648 | LCC = 0.8550 | SRCC = 0.8547 ]
|
| 35 |
+
INFO:root:Average loss=0.14323040499137
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2619 | LCC = 0.8552 | SRCC = 0.8540 ]
|
| 38 |
+
INFO:root:Average loss=0.13883352661744142
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2436 | LCC = 0.8555 | SRCC = 0.8536 ]
|
| 41 |
+
INFO:root:Average loss=0.13759343230571502
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2495 | LCC = 0.8558 | SRCC = 0.8549 ]
|
| 44 |
+
INFO:root:Average loss=0.12571560648771432
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2366 | LCC = 0.8557 | SRCC = 0.8539 ]
|
| 47 |
+
INFO:root:Average loss=0.11170433576290424
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2476 | LCC = 0.8527 | SRCC = 0.8527 ]
|
| 50 |
+
INFO:root:Average loss=0.11621865797310303
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2430 | LCC = 0.8506 | SRCC = 0.8492 ]
|
| 53 |
+
INFO:root:Average loss=0.11055554239413677
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2520 | LCC = 0.8499 | SRCC = 0.8491 ]
|
| 56 |
+
INFO:root:Average loss=0.09552048122844635
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2442 | LCC = 0.8482 | SRCC = 0.8477 ]
|
| 59 |
+
INFO:root:Average loss=0.0940097991663676
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2473 | LCC = 0.8450 | SRCC = 0.8442 ]
|
| 62 |
+
INFO:root:Average loss=0.08238960525546318
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2469 | LCC = 0.8447 | SRCC = 0.8448 ]
|
| 65 |
+
INFO:root:Average loss=0.08315739842752616
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2684 | LCC = 0.8465 | SRCC = 0.8446 ]
|
| 68 |
+
INFO:root:Average loss=0.07951096015480849
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2539 | LCC = 0.8463 | SRCC = 0.8448 ]
|
| 71 |
+
INFO:root:Average loss=0.0808035160582035
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2467 | LCC = 0.8466 | SRCC = 0.8462 ]
|
| 74 |
+
INFO:root:Average loss=0.06967812523436852
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2423 | LCC = 0.8485 | SRCC = 0.8481 ]
|
| 77 |
+
INFO:root:Average loss=0.06693637542999707
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2659 | LCC = 0.8458 | SRCC = 0.8452 ]
|
| 80 |
+
INFO:root:Average loss=0.06728526051992025
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2427 | LCC = 0.8468 | SRCC = 0.8467 ]
|
| 83 |
+
INFO:root:Average loss=0.06428273881857212
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2512 | LCC = 0.8420 | SRCC = 0.8412 ]
|
| 86 |
+
INFO:root:Average loss=0.06212889455640928
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2510 | LCC = 0.8475 | SRCC = 0.8468 ]
|
| 89 |
+
INFO:root:Average loss=0.05527650297452242
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2555 | LCC = 0.8418 | SRCC = 0.8410 ]
|
| 92 |
+
INFO:root:Average loss=0.05925052374219283
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2549 | LCC = 0.8511 | SRCC = 0.8518 ]
|
| 95 |
+
INFO:root:Average loss=0.0530470193674167
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2690 | LCC = 0.8456 | SRCC = 0.8437 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2736 | LCC = 0.8476 | SRCC = 0.8479 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/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_2epoch/layer7/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c005cc74d5043872a925bfe06ff3108e12e70282024c0b9cb22bc4a514fe008
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9e46e158aa41028fdf4ecd033cb845311dcaf0fafaf73575ad130afc911de705
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/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.452056314318608
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3180 | LCC = 0.8033 | SRCC = 0.8126 ]
|
| 8 |
+
INFO:root:Average loss=0.28075962494581175
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3133 | LCC = 0.8388 | SRCC = 0.8359 ]
|
| 11 |
+
INFO:root:Average loss=0.24883588747336313
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2865 | LCC = 0.8490 | SRCC = 0.8459 ]
|
| 14 |
+
INFO:root:Average loss=0.22087921412327352
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2560 | LCC = 0.8562 | SRCC = 0.8534 ]
|
| 17 |
+
INFO:root:Average loss=0.20980743165963736
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2687 | LCC = 0.8581 | SRCC = 0.8557 ]
|
| 20 |
+
INFO:root:Average loss=0.19847632696231207
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2448 | LCC = 0.8600 | SRCC = 0.8589 ]
|
| 23 |
+
INFO:root:Average loss=0.17746730502217245
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2413 | LCC = 0.8595 | SRCC = 0.8583 ]
|
| 26 |
+
INFO:root:Average loss=0.16760278779726762
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2404 | LCC = 0.8605 | SRCC = 0.8585 ]
|
| 29 |
+
INFO:root:Average loss=0.15927141522749877
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2363 | LCC = 0.8587 | SRCC = 0.8557 ]
|
| 32 |
+
INFO:root:Average loss=0.14369378086083975
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2501 | LCC = 0.8568 | SRCC = 0.8550 ]
|
| 35 |
+
INFO:root:Average loss=0.13832736101288062
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2567 | LCC = 0.8544 | SRCC = 0.8524 ]
|
| 38 |
+
INFO:root:Average loss=0.12984634582430887
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2514 | LCC = 0.8541 | SRCC = 0.8525 ]
|
| 41 |
+
INFO:root:Average loss=0.11912940853299239
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2415 | LCC = 0.8569 | SRCC = 0.8557 ]
|
| 44 |
+
INFO:root:Average loss=0.11701363540039612
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2581 | LCC = 0.8531 | SRCC = 0.8513 ]
|
| 47 |
+
INFO:root:Average loss=0.10647028144926597
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2474 | LCC = 0.8503 | SRCC = 0.8478 ]
|
| 50 |
+
INFO:root:Average loss=0.09998841378360222
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2427 | LCC = 0.8517 | SRCC = 0.8514 ]
|
| 53 |
+
INFO:root:Average loss=0.09776374286948106
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2416 | LCC = 0.8488 | SRCC = 0.8487 ]
|
| 56 |
+
INFO:root:Average loss=0.09849136690489757
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2510 | LCC = 0.8530 | SRCC = 0.8514 ]
|
| 59 |
+
INFO:root:Average loss=0.08787564326746342
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2531 | LCC = 0.8468 | SRCC = 0.8472 ]
|
| 62 |
+
INFO:root:Average loss=0.08047295696078202
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2503 | LCC = 0.8481 | SRCC = 0.8469 ]
|
| 65 |
+
INFO:root:Average loss=0.0800807113544299
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2806 | LCC = 0.8476 | SRCC = 0.8461 ]
|
| 68 |
+
INFO:root:Average loss=0.07766213654898681
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2480 | LCC = 0.8466 | SRCC = 0.8454 ]
|
| 71 |
+
INFO:root:Average loss=0.06867456698837952
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2421 | LCC = 0.8462 | SRCC = 0.8455 ]
|
| 74 |
+
INFO:root:Average loss=0.07454978598233981
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2873 | LCC = 0.8408 | SRCC = 0.8422 ]
|
| 77 |
+
INFO:root:Average loss=0.06894614456746823
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2485 | LCC = 0.8482 | SRCC = 0.8472 ]
|
| 80 |
+
INFO:root:Average loss=0.06405419498108901
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2435 | LCC = 0.8488 | SRCC = 0.8475 ]
|
| 83 |
+
INFO:root:Average loss=0.06023827629784743
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2630 | LCC = 0.8463 | SRCC = 0.8452 ]
|
| 86 |
+
INFO:root:Average loss=0.05772898284097513
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2517 | LCC = 0.8441 | SRCC = 0.8437 ]
|
| 89 |
+
INFO:root:Average loss=0.05193462480719273
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2475 | LCC = 0.8469 | SRCC = 0.8466 ]
|
| 92 |
+
INFO:root:Average loss=0.05206845416568029
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2494 | LCC = 0.8462 | SRCC = 0.8451 ]
|
| 95 |
+
INFO:root:Average loss=0.048558820803196005
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2480 | LCC = 0.8450 | SRCC = 0.8443 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2674 | LCC = 0.8480 | SRCC = 0.8486 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a75ea57d1c60fcc537a34a6c7f3689c31eaec88efc4140bafdc7bfb668461c43
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2cffb8133bf1e41da74124aac9cb28a7eb05e9d22f8f5ddda8411c7e55d06b40
|
| 3 |
+
size 2749638
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.47223941676127607
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3751 | LCC = 0.7904 | SRCC = 0.8011 ]
|
| 8 |
+
INFO:root:Average loss=0.27982731278126055
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3302 | LCC = 0.8303 | SRCC = 0.8308 ]
|
| 11 |
+
INFO:root:Average loss=0.2549691217449995
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2838 | LCC = 0.8463 | SRCC = 0.8453 ]
|
| 14 |
+
INFO:root:Average loss=0.23500471390210664
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2539 | LCC = 0.8503 | SRCC = 0.8517 ]
|
| 17 |
+
INFO:root:Average loss=0.21987843398864454
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2718 | LCC = 0.8506 | SRCC = 0.8522 ]
|
| 20 |
+
INFO:root:Average loss=0.2170359866741376
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2457 | LCC = 0.8586 | SRCC = 0.8584 ]
|
| 23 |
+
INFO:root:Average loss=0.20060518842477065
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2286 | LCC = 0.8626 | SRCC = 0.8630 ]
|
| 26 |
+
INFO:root:Average loss=0.19284183226334742
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2629 | LCC = 0.8586 | SRCC = 0.8582 ]
|
| 29 |
+
INFO:root:Average loss=0.17678190242403594
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2560 | LCC = 0.8595 | SRCC = 0.8614 ]
|
| 32 |
+
INFO:root:Average loss=0.17932205007244378
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2733 | LCC = 0.8643 | SRCC = 0.8641 ]
|
| 35 |
+
INFO:root:Average loss=0.16425106836817202
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2289 | LCC = 0.8613 | SRCC = 0.8604 ]
|
| 38 |
+
INFO:root:Average loss=0.16195728620275474
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2871 | LCC = 0.8548 | SRCC = 0.8567 ]
|
| 41 |
+
INFO:root:Average loss=0.14901328545350295
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2531 | LCC = 0.8595 | SRCC = 0.8583 ]
|
| 44 |
+
INFO:root:Average loss=0.1450820328333439
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2588 | LCC = 0.8582 | SRCC = 0.8622 ]
|
| 47 |
+
INFO:root:Average loss=0.14872629548876715
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2499 | LCC = 0.8616 | SRCC = 0.8629 ]
|
| 50 |
+
INFO:root:Average loss=0.13497571045389542
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2722 | LCC = 0.8587 | SRCC = 0.8610 ]
|
| 53 |
+
INFO:root:Average loss=0.12823205307508126
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2513 | LCC = 0.8575 | SRCC = 0.8599 ]
|
| 56 |
+
INFO:root:Average loss=0.11853662391121571
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2392 | LCC = 0.8583 | SRCC = 0.8596 ]
|
| 59 |
+
INFO:root:Average loss=0.1215384110617332
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2400 | LCC = 0.8624 | SRCC = 0.8627 ]
|
| 62 |
+
INFO:root:Average loss=0.11138051327986595
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2692 | LCC = 0.8510 | SRCC = 0.8559 ]
|
| 65 |
+
INFO:root:Average loss=0.10720421330860028
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2219 | LCC = 0.8603 | SRCC = 0.8609 ]
|
| 68 |
+
INFO:root:Average loss=0.10368176158039998
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2606 | LCC = 0.8563 | SRCC = 0.8613 ]
|
| 71 |
+
INFO:root:Average loss=0.09395950932342273
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2329 | LCC = 0.8539 | SRCC = 0.8549 ]
|
| 74 |
+
INFO:root:Average loss=0.0912096606901823
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2483 | LCC = 0.8518 | SRCC = 0.8572 ]
|
| 77 |
+
INFO:root:Average loss=0.0967549738020469
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2274 | LCC = 0.8592 | SRCC = 0.8603 ]
|
| 80 |
+
INFO:root:Average loss=0.08598366322425696
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2381 | LCC = 0.8554 | SRCC = 0.8547 ]
|
| 83 |
+
INFO:root:Average loss=0.08496240774790446
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2299 | LCC = 0.8567 | SRCC = 0.8560 ]
|
| 86 |
+
INFO:root:Average loss=0.07885864548958264
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2292 | LCC = 0.8565 | SRCC = 0.8591 ]
|
| 89 |
+
INFO:root:Average loss=0.07814654598060326
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2370 | LCC = 0.8523 | SRCC = 0.8532 ]
|
| 92 |
+
INFO:root:Average loss=0.0762549732835629
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2385 | LCC = 0.8525 | SRCC = 0.8537 ]
|
| 95 |
+
INFO:root:Average loss=0.07473513718025807
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2510 | LCC = 0.8476 | SRCC = 0.8498 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2667 | LCC = 0.8646 | SRCC = 0.8623 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fdafd90b7bc12c729c4df2e3a1609d1346e0c4c2e5ebf3bd6b18b0c23d7c7de3
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9dd2f52000cdb89be8a5b10597c15b96ad3e0745365a10c38fbe2eb634132fef
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.41303758246776384
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3732 | LCC = 0.8069 | SRCC = 0.8122 ]
|
| 8 |
+
INFO:root:Average loss=0.285670280838624
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3252 | LCC = 0.8318 | SRCC = 0.8336 ]
|
| 11 |
+
INFO:root:Average loss=0.2465732741432312
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2782 | LCC = 0.8431 | SRCC = 0.8426 ]
|
| 14 |
+
INFO:root:Average loss=0.23031453559031853
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2695 | LCC = 0.8469 | SRCC = 0.8471 ]
|
| 17 |
+
INFO:root:Average loss=0.2165564417074888
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2533 | LCC = 0.8569 | SRCC = 0.8579 ]
|
| 20 |
+
INFO:root:Average loss=0.20914513503129667
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2484 | LCC = 0.8551 | SRCC = 0.8553 ]
|
| 23 |
+
INFO:root:Average loss=0.19378782894748908
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2830 | LCC = 0.8554 | SRCC = 0.8556 ]
|
| 26 |
+
INFO:root:Average loss=0.19045280875303808
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2577 | LCC = 0.8613 | SRCC = 0.8614 ]
|
| 29 |
+
INFO:root:Average loss=0.1825267035418596
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2559 | LCC = 0.8612 | SRCC = 0.8587 ]
|
| 32 |
+
INFO:root:Average loss=0.1781048531142565
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2480 | LCC = 0.8628 | SRCC = 0.8614 ]
|
| 35 |
+
INFO:root:Average loss=0.1567079455424578
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2610 | LCC = 0.8596 | SRCC = 0.8582 ]
|
| 38 |
+
INFO:root:Average loss=0.15464371643387353
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2663 | LCC = 0.8602 | SRCC = 0.8593 ]
|
| 41 |
+
INFO:root:Average loss=0.15943133668639722
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2434 | LCC = 0.8588 | SRCC = 0.8588 ]
|
| 44 |
+
INFO:root:Average loss=0.13974300399422646
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2527 | LCC = 0.8597 | SRCC = 0.8590 ]
|
| 47 |
+
INFO:root:Average loss=0.13362917944024771
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2506 | LCC = 0.8576 | SRCC = 0.8567 ]
|
| 50 |
+
INFO:root:Average loss=0.14108931206357786
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2575 | LCC = 0.8563 | SRCC = 0.8579 ]
|
| 53 |
+
INFO:root:Average loss=0.12080439170583701
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2339 | LCC = 0.8572 | SRCC = 0.8572 ]
|
| 56 |
+
INFO:root:Average loss=0.12141439557457581
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2522 | LCC = 0.8558 | SRCC = 0.8563 ]
|
| 59 |
+
INFO:root:Average loss=0.11084135382985458
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2263 | LCC = 0.8595 | SRCC = 0.8601 ]
|
| 62 |
+
INFO:root:Average loss=0.10768489721111763
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2399 | LCC = 0.8555 | SRCC = 0.8557 ]
|
| 65 |
+
INFO:root:Average loss=0.10416071470349263
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2568 | LCC = 0.8535 | SRCC = 0.8535 ]
|
| 68 |
+
INFO:root:Average loss=0.1003027319526061
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2479 | LCC = 0.8515 | SRCC = 0.8523 ]
|
| 71 |
+
INFO:root:Average loss=0.09413785067124245
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2455 | LCC = 0.8547 | SRCC = 0.8538 ]
|
| 74 |
+
INFO:root:Average loss=0.09474462060592113
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2563 | LCC = 0.8475 | SRCC = 0.8485 ]
|
| 77 |
+
INFO:root:Average loss=0.08715413195582536
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2635 | LCC = 0.8497 | SRCC = 0.8499 ]
|
| 80 |
+
INFO:root:Average loss=0.08358084114316182
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2458 | LCC = 0.8492 | SRCC = 0.8482 ]
|
| 83 |
+
INFO:root:Average loss=0.08539289790086257
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2477 | LCC = 0.8525 | SRCC = 0.8543 ]
|
| 86 |
+
INFO:root:Average loss=0.07988832332193851
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2406 | LCC = 0.8528 | SRCC = 0.8537 ]
|
| 89 |
+
INFO:root:Average loss=0.0792451973240345
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2300 | LCC = 0.8561 | SRCC = 0.8560 ]
|
| 92 |
+
INFO:root:Average loss=0.07189440622161596
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2456 | LCC = 0.8487 | SRCC = 0.8490 ]
|
| 95 |
+
INFO:root:Average loss=0.07597707555844234
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2339 | LCC = 0.8544 | SRCC = 0.8541 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2507 | LCC = 0.8575 | SRCC = 0.8566 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:28554270ea3746945a70254d3c16fa1891e297fe95b35782b784777732203d60
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ab01059d7aa73d848a850fac307178a448ea569cade2e633670f4f4112b46571
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.4284714019069305
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4288 | LCC = 0.8063 | SRCC = 0.8101 ]
|
| 8 |
+
INFO:root:Average loss=0.2851868722683344
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3220 | LCC = 0.8339 | SRCC = 0.8350 ]
|
| 11 |
+
INFO:root:Average loss=0.24797301987806955
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3159 | LCC = 0.8416 | SRCC = 0.8408 ]
|
| 14 |
+
INFO:root:Average loss=0.23648603585286018
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3584 | LCC = 0.8436 | SRCC = 0.8444 ]
|
| 17 |
+
INFO:root:Average loss=0.21922305016181406
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3090 | LCC = 0.8522 | SRCC = 0.8512 ]
|
| 20 |
+
INFO:root:Average loss=0.23053583999474844
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2908 | LCC = 0.8516 | SRCC = 0.8494 ]
|
| 23 |
+
INFO:root:Average loss=0.20694475363080317
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2505 | LCC = 0.8596 | SRCC = 0.8567 ]
|
| 26 |
+
INFO:root:Average loss=0.1895792339092646
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2652 | LCC = 0.8616 | SRCC = 0.8609 ]
|
| 29 |
+
INFO:root:Average loss=0.17921728335129908
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2995 | LCC = 0.8570 | SRCC = 0.8550 ]
|
| 32 |
+
INFO:root:Average loss=0.16879250319340292
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2540 | LCC = 0.8542 | SRCC = 0.8517 ]
|
| 35 |
+
INFO:root:Average loss=0.16210990771651268
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2948 | LCC = 0.8568 | SRCC = 0.8579 ]
|
| 38 |
+
INFO:root:Average loss=0.15975914952846673
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3007 | LCC = 0.8533 | SRCC = 0.8534 ]
|
| 41 |
+
INFO:root:Average loss=0.1476324700201169
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2552 | LCC = 0.8567 | SRCC = 0.8565 ]
|
| 44 |
+
INFO:root:Average loss=0.13965487948212868
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2529 | LCC = 0.8545 | SRCC = 0.8533 ]
|
| 47 |
+
INFO:root:Average loss=0.1314195753672184
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2787 | LCC = 0.8546 | SRCC = 0.8535 ]
|
| 50 |
+
INFO:root:Average loss=0.12969911471009254
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2740 | LCC = 0.8548 | SRCC = 0.8525 ]
|
| 53 |
+
INFO:root:Average loss=0.12821087948022744
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2554 | LCC = 0.8507 | SRCC = 0.8486 ]
|
| 56 |
+
INFO:root:Average loss=0.12309678414693245
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2384 | LCC = 0.8510 | SRCC = 0.8487 ]
|
| 59 |
+
INFO:root:Average loss=0.10895248215932113
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3008 | LCC = 0.8464 | SRCC = 0.8450 ]
|
| 62 |
+
INFO:root:Average loss=0.10654697099175209
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2488 | LCC = 0.8468 | SRCC = 0.8454 ]
|
| 65 |
+
INFO:root:Average loss=0.1012338790564965
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3068 | LCC = 0.8422 | SRCC = 0.8417 ]
|
| 68 |
+
INFO:root:Average loss=0.10265656675283726
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2414 | LCC = 0.8506 | SRCC = 0.8495 ]
|
| 71 |
+
INFO:root:Average loss=0.0958816347978054
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2623 | LCC = 0.8450 | SRCC = 0.8436 ]
|
| 74 |
+
INFO:root:Average loss=0.08844794638645954
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2644 | LCC = 0.8445 | SRCC = 0.8442 ]
|
| 77 |
+
INFO:root:Average loss=0.08726239767976296
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2722 | LCC = 0.8456 | SRCC = 0.8444 ]
|
| 80 |
+
INFO:root:Average loss=0.08903845958411694
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3081 | LCC = 0.8481 | SRCC = 0.8483 ]
|
| 83 |
+
INFO:root:Average loss=0.08786079831994496
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2508 | LCC = 0.8497 | SRCC = 0.8485 ]
|
| 86 |
+
INFO:root:Average loss=0.08469467180279586
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2450 | LCC = 0.8485 | SRCC = 0.8476 ]
|
| 89 |
+
INFO:root:Average loss=0.07920091355649325
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2528 | LCC = 0.8470 | SRCC = 0.8465 ]
|
| 92 |
+
INFO:root:Average loss=0.07507136316062549
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2453 | LCC = 0.8480 | SRCC = 0.8474 ]
|
| 95 |
+
INFO:root:Average loss=0.0728788711130619
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2559 | LCC = 0.8486 | SRCC = 0.8482 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2579 | LCC = 0.8554 | SRCC = 0.8532 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39e04901f0c7684dec0f71c23996942b5de1561e7b298e98635d26038f918778
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:817ef44e113973ca2620964878941835dd0cddb7070ab69523e6da0414097fc5
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.40541908622552186
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3292 | LCC = 0.8130 | SRCC = 0.8154 ]
|
| 8 |
+
INFO:root:Average loss=0.27299155103854644
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2932 | LCC = 0.8382 | SRCC = 0.8358 ]
|
| 11 |
+
INFO:root:Average loss=0.2485059251387914
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2642 | LCC = 0.8462 | SRCC = 0.8457 ]
|
| 14 |
+
INFO:root:Average loss=0.2358058819021934
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2710 | LCC = 0.8525 | SRCC = 0.8533 ]
|
| 17 |
+
INFO:root:Average loss=0.2178884758017002
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2632 | LCC = 0.8556 | SRCC = 0.8525 ]
|
| 20 |
+
INFO:root:Average loss=0.21346342353484568
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2489 | LCC = 0.8528 | SRCC = 0.8503 ]
|
| 23 |
+
INFO:root:Average loss=0.20036793786745805
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2706 | LCC = 0.8604 | SRCC = 0.8579 ]
|
| 26 |
+
INFO:root:Average loss=0.18231598717662004
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2438 | LCC = 0.8625 | SRCC = 0.8604 ]
|
| 29 |
+
INFO:root:Average loss=0.17464523705152366
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2510 | LCC = 0.8567 | SRCC = 0.8542 ]
|
| 32 |
+
INFO:root:Average loss=0.16686499223877221
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2490 | LCC = 0.8642 | SRCC = 0.8619 ]
|
| 35 |
+
INFO:root:Average loss=0.15523532529671988
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2472 | LCC = 0.8595 | SRCC = 0.8564 ]
|
| 38 |
+
INFO:root:Average loss=0.14768393203998223
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2777 | LCC = 0.8592 | SRCC = 0.8579 ]
|
| 41 |
+
INFO:root:Average loss=0.14827872946476325
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2755 | LCC = 0.8584 | SRCC = 0.8579 ]
|
| 44 |
+
INFO:root:Average loss=0.1361596627304187
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2317 | LCC = 0.8554 | SRCC = 0.8531 ]
|
| 47 |
+
INFO:root:Average loss=0.12682292954279825
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2496 | LCC = 0.8557 | SRCC = 0.8531 ]
|
| 50 |
+
INFO:root:Average loss=0.12293366266366763
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2605 | LCC = 0.8584 | SRCC = 0.8576 ]
|
| 53 |
+
INFO:root:Average loss=0.11543400318194659
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2605 | LCC = 0.8524 | SRCC = 0.8505 ]
|
| 56 |
+
INFO:root:Average loss=0.11308515215149292
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2459 | LCC = 0.8544 | SRCC = 0.8535 ]
|
| 59 |
+
INFO:root:Average loss=0.11106220088326015
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2462 | LCC = 0.8513 | SRCC = 0.8499 ]
|
| 62 |
+
INFO:root:Average loss=0.10959643316574585
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2388 | LCC = 0.8552 | SRCC = 0.8537 ]
|
| 65 |
+
INFO:root:Average loss=0.10280151264025615
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2367 | LCC = 0.8572 | SRCC = 0.8561 ]
|
| 68 |
+
INFO:root:Average loss=0.09583375765345035
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2472 | LCC = 0.8525 | SRCC = 0.8525 ]
|
| 71 |
+
INFO:root:Average loss=0.09108715327695394
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2474 | LCC = 0.8538 | SRCC = 0.8519 ]
|
| 74 |
+
INFO:root:Average loss=0.08602023435135682
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2464 | LCC = 0.8517 | SRCC = 0.8498 ]
|
| 77 |
+
INFO:root:Average loss=0.08819783736880009
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2689 | LCC = 0.8464 | SRCC = 0.8452 ]
|
| 80 |
+
INFO:root:Average loss=0.0810032367037657
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2375 | LCC = 0.8509 | SRCC = 0.8496 ]
|
| 83 |
+
INFO:root:Average loss=0.07612811587750912
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2506 | LCC = 0.8500 | SRCC = 0.8486 ]
|
| 86 |
+
INFO:root:Average loss=0.08309972878450002
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2633 | LCC = 0.8494 | SRCC = 0.8484 ]
|
| 89 |
+
INFO:root:Average loss=0.07546157108094448
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2501 | LCC = 0.8498 | SRCC = 0.8478 ]
|
| 92 |
+
INFO:root:Average loss=0.0741928331553936
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2582 | LCC = 0.8511 | SRCC = 0.8502 ]
|
| 95 |
+
INFO:root:Average loss=0.06476368224964692
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2410 | LCC = 0.8492 | SRCC = 0.8491 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2692 | LCC = 0.8569 | SRCC = 0.8593 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a7bbb8e0472ccf1fbd917353cb8eda0f47b45934cd5cfc6aeee445c8fe214b0
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b0f3f4e414f8b12d9d5dcda1231a614be423c83892f5772f68e636a1dd89c78
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.43741939942806196
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3538 | LCC = 0.7955 | SRCC = 0.7995 ]
|
| 8 |
+
INFO:root:Average loss=0.2892684185734162
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3225 | LCC = 0.8224 | SRCC = 0.8260 ]
|
| 11 |
+
INFO:root:Average loss=0.24687125247258407
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2957 | LCC = 0.8453 | SRCC = 0.8444 ]
|
| 14 |
+
INFO:root:Average loss=0.22453046322633058
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2956 | LCC = 0.8495 | SRCC = 0.8497 ]
|
| 17 |
+
INFO:root:Average loss=0.2161407819352089
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2601 | LCC = 0.8594 | SRCC = 0.8583 ]
|
| 20 |
+
INFO:root:Average loss=0.2075838952874526
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2637 | LCC = 0.8630 | SRCC = 0.8620 ]
|
| 23 |
+
INFO:root:Average loss=0.2106938322003071
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2440 | LCC = 0.8615 | SRCC = 0.8610 ]
|
| 26 |
+
INFO:root:Average loss=0.1833732172082632
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2426 | LCC = 0.8628 | SRCC = 0.8626 ]
|
| 29 |
+
INFO:root:Average loss=0.17070448857087356
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2519 | LCC = 0.8630 | SRCC = 0.8629 ]
|
| 32 |
+
INFO:root:Average loss=0.16309963032985345
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2384 | LCC = 0.8641 | SRCC = 0.8645 ]
|
| 35 |
+
INFO:root:Average loss=0.15788956072468024
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2264 | LCC = 0.8653 | SRCC = 0.8652 ]
|
| 38 |
+
INFO:root:Average loss=0.15759127290967184
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3060 | LCC = 0.8598 | SRCC = 0.8622 ]
|
| 41 |
+
INFO:root:Average loss=0.15851521339171973
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2240 | LCC = 0.8663 | SRCC = 0.8675 ]
|
| 44 |
+
INFO:root:Average loss=0.1394150136755063
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2274 | LCC = 0.8664 | SRCC = 0.8670 ]
|
| 47 |
+
INFO:root:Average loss=0.13565322651694983
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2362 | LCC = 0.8644 | SRCC = 0.8649 ]
|
| 50 |
+
INFO:root:Average loss=0.12542991130015788
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2336 | LCC = 0.8641 | SRCC = 0.8647 ]
|
| 53 |
+
INFO:root:Average loss=0.11608249178299537
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2440 | LCC = 0.8606 | SRCC = 0.8614 ]
|
| 56 |
+
INFO:root:Average loss=0.11595186696220668
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2413 | LCC = 0.8592 | SRCC = 0.8591 ]
|
| 59 |
+
INFO:root:Average loss=0.11024917996464631
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2382 | LCC = 0.8596 | SRCC = 0.8605 ]
|
| 62 |
+
INFO:root:Average loss=0.10416477918624878
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2674 | LCC = 0.8542 | SRCC = 0.8535 ]
|
| 65 |
+
INFO:root:Average loss=0.09718882345045224
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2379 | LCC = 0.8586 | SRCC = 0.8599 ]
|
| 68 |
+
INFO:root:Average loss=0.09196937079421985
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2561 | LCC = 0.8534 | SRCC = 0.8541 ]
|
| 71 |
+
INFO:root:Average loss=0.09043105338246395
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2331 | LCC = 0.8574 | SRCC = 0.8587 ]
|
| 74 |
+
INFO:root:Average loss=0.08995892900304917
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2339 | LCC = 0.8598 | SRCC = 0.8597 ]
|
| 77 |
+
INFO:root:Average loss=0.08405351543273681
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2510 | LCC = 0.8571 | SRCC = 0.8573 ]
|
| 80 |
+
INFO:root:Average loss=0.07768201933075221
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2522 | LCC = 0.8562 | SRCC = 0.8572 ]
|
| 83 |
+
INFO:root:Average loss=0.0780353932044445
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2251 | LCC = 0.8571 | SRCC = 0.8581 ]
|
| 86 |
+
INFO:root:Average loss=0.07864247553814681
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2314 | LCC = 0.8559 | SRCC = 0.8565 ]
|
| 89 |
+
INFO:root:Average loss=0.07323050661346851
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2243 | LCC = 0.8572 | SRCC = 0.8585 ]
|
| 92 |
+
INFO:root:Average loss=0.07265138511474316
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2550 | LCC = 0.8526 | SRCC = 0.8534 ]
|
| 95 |
+
INFO:root:Average loss=0.0646772900930582
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2387 | LCC = 0.8538 | SRCC = 0.8545 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2626 | LCC = 0.8533 | SRCC = 0.8526 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8712c99bb3cbfe5a0a5e91f3d2054ea82f6084c0e6ed874a46179e4a3bf6dca8
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:87bc07c53d1a2118dfc2a1a194b372b56b536e961461d15da3d0c288e1571401
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.40266313346532673
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3359 | LCC = 0.7966 | SRCC = 0.7982 ]
|
| 8 |
+
INFO:root:Average loss=0.29929162313540775
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3918 | LCC = 0.8220 | SRCC = 0.8244 ]
|
| 11 |
+
INFO:root:Average loss=0.2818653656122012
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3207 | LCC = 0.8372 | SRCC = 0.8360 ]
|
| 14 |
+
INFO:root:Average loss=0.263999010125796
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2658 | LCC = 0.8363 | SRCC = 0.8379 ]
|
| 17 |
+
INFO:root:Average loss=0.2569745114216438
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3762 | LCC = 0.8436 | SRCC = 0.8459 ]
|
| 20 |
+
INFO:root:Average loss=0.24703097190612403
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2841 | LCC = 0.8474 | SRCC = 0.8477 ]
|
| 23 |
+
INFO:root:Average loss=0.2624498519759912
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3735 | LCC = 0.8507 | SRCC = 0.8502 ]
|
| 26 |
+
INFO:root:Average loss=0.23869414455615556
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2764 | LCC = 0.8506 | SRCC = 0.8507 ]
|
| 29 |
+
INFO:root:Average loss=0.21669101361662912
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2513 | LCC = 0.8581 | SRCC = 0.8578 ]
|
| 32 |
+
INFO:root:Average loss=0.20718366289750123
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3034 | LCC = 0.8592 | SRCC = 0.8612 ]
|
| 35 |
+
INFO:root:Average loss=0.2161941912311774
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3129 | LCC = 0.8583 | SRCC = 0.8575 ]
|
| 38 |
+
INFO:root:Average loss=0.21269655743470559
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3040 | LCC = 0.8601 | SRCC = 0.8583 ]
|
| 41 |
+
INFO:root:Average loss=0.20141950383400306
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2737 | LCC = 0.8605 | SRCC = 0.8580 ]
|
| 44 |
+
INFO:root:Average loss=0.20354594595921346
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2812 | LCC = 0.8585 | SRCC = 0.8570 ]
|
| 47 |
+
INFO:root:Average loss=0.19188268568653327
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3086 | LCC = 0.8573 | SRCC = 0.8610 ]
|
| 50 |
+
INFO:root:Average loss=0.1888957713276912
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3459 | LCC = 0.8562 | SRCC = 0.8609 ]
|
| 53 |
+
INFO:root:Average loss=0.18730529254445663
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2625 | LCC = 0.8589 | SRCC = 0.8570 ]
|
| 56 |
+
INFO:root:Average loss=0.1779670580648459
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2522 | LCC = 0.8563 | SRCC = 0.8567 ]
|
| 59 |
+
INFO:root:Average loss=0.1712119739789229
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2592 | LCC = 0.8588 | SRCC = 0.8585 ]
|
| 62 |
+
INFO:root:Average loss=0.17276713022818932
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2630 | LCC = 0.8600 | SRCC = 0.8582 ]
|
| 65 |
+
INFO:root:Average loss=0.1593610692100647
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2969 | LCC = 0.8575 | SRCC = 0.8592 ]
|
| 68 |
+
INFO:root:Average loss=0.1659397159058314
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2753 | LCC = 0.8618 | SRCC = 0.8607 ]
|
| 71 |
+
INFO:root:Average loss=0.1549930579196184
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2566 | LCC = 0.8619 | SRCC = 0.8616 ]
|
| 74 |
+
INFO:root:Average loss=0.16745337662406456
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2867 | LCC = 0.8570 | SRCC = 0.8570 ]
|
| 77 |
+
INFO:root:Average loss=0.15673001817403695
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3009 | LCC = 0.8588 | SRCC = 0.8578 ]
|
| 80 |
+
INFO:root:Average loss=0.14925719224489653
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2394 | LCC = 0.8603 | SRCC = 0.8587 ]
|
| 83 |
+
INFO:root:Average loss=0.15450562527164435
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2645 | LCC = 0.8614 | SRCC = 0.8594 ]
|
| 86 |
+
INFO:root:Average loss=0.1366595337406183
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2373 | LCC = 0.8575 | SRCC = 0.8565 ]
|
| 89 |
+
INFO:root:Average loss=0.13738801101079354
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2639 | LCC = 0.8533 | SRCC = 0.8522 ]
|
| 92 |
+
INFO:root:Average loss=0.1267202989413188
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2985 | LCC = 0.8510 | SRCC = 0.8510 ]
|
| 95 |
+
INFO:root:Average loss=0.12687058336077592
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2258 | LCC = 0.8599 | SRCC = 0.8586 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2939 | LCC = 0.8493 | SRCC = 0.8522 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c236fa89aa193ebad1e250f365e8baee945fb63a83c1f20b9871743db6e32abb
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f35b472145f937f13071729146a86316d7127143bfacc1e8e7a56a24b259c011
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.4648682929766484
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3894 | LCC = 0.7893 | SRCC = 0.7919 ]
|
| 8 |
+
INFO:root:Average loss=0.29751755621952886
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3217 | LCC = 0.8200 | SRCC = 0.8215 ]
|
| 11 |
+
INFO:root:Average loss=0.2782701930174461
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3154 | LCC = 0.8149 | SRCC = 0.8217 ]
|
| 14 |
+
INFO:root:Average loss=0.2735561869847469
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3193 | LCC = 0.8401 | SRCC = 0.8377 ]
|
| 17 |
+
INFO:root:Average loss=0.26198548556138307
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3044 | LCC = 0.8434 | SRCC = 0.8425 ]
|
| 20 |
+
INFO:root:Average loss=0.24024102378350037
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2920 | LCC = 0.8517 | SRCC = 0.8515 ]
|
| 23 |
+
INFO:root:Average loss=0.2412498067968931
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2637 | LCC = 0.8512 | SRCC = 0.8502 ]
|
| 26 |
+
INFO:root:Average loss=0.24265832874255303
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3037 | LCC = 0.8526 | SRCC = 0.8512 ]
|
| 29 |
+
INFO:root:Average loss=0.2187001244762005
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2693 | LCC = 0.8561 | SRCC = 0.8561 ]
|
| 32 |
+
INFO:root:Average loss=0.2090743953982989
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2790 | LCC = 0.8521 | SRCC = 0.8535 ]
|
| 35 |
+
INFO:root:Average loss=0.20778891501518396
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2772 | LCC = 0.8602 | SRCC = 0.8603 ]
|
| 38 |
+
INFO:root:Average loss=0.21923466695424837
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2922 | LCC = 0.8600 | SRCC = 0.8575 ]
|
| 41 |
+
INFO:root:Average loss=0.19139932201076776
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3009 | LCC = 0.8639 | SRCC = 0.8625 ]
|
| 44 |
+
INFO:root:Average loss=0.19710553341951126
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3082 | LCC = 0.8659 | SRCC = 0.8639 ]
|
| 47 |
+
INFO:root:Average loss=0.18678924231193003
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2350 | LCC = 0.8624 | SRCC = 0.8617 ]
|
| 50 |
+
INFO:root:Average loss=0.18032619930230653
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2665 | LCC = 0.8625 | SRCC = 0.8607 ]
|
| 53 |
+
INFO:root:Average loss=0.1813369193711342
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2409 | LCC = 0.8636 | SRCC = 0.8624 ]
|
| 56 |
+
INFO:root:Average loss=0.16888423808492148
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2555 | LCC = 0.8607 | SRCC = 0.8598 ]
|
| 59 |
+
INFO:root:Average loss=0.17025642154308465
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2545 | LCC = 0.8646 | SRCC = 0.8632 ]
|
| 62 |
+
INFO:root:Average loss=0.16329536577447867
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2634 | LCC = 0.8619 | SRCC = 0.8599 ]
|
| 65 |
+
INFO:root:Average loss=0.16247353483086976
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2443 | LCC = 0.8630 | SRCC = 0.8624 ]
|
| 68 |
+
INFO:root:Average loss=0.160555169177361
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2394 | LCC = 0.8619 | SRCC = 0.8615 ]
|
| 71 |
+
INFO:root:Average loss=0.163032844185065
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2617 | LCC = 0.8571 | SRCC = 0.8571 ]
|
| 74 |
+
INFO:root:Average loss=0.15858055183138603
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2323 | LCC = 0.8579 | SRCC = 0.8590 ]
|
| 77 |
+
INFO:root:Average loss=0.14591626756084272
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2573 | LCC = 0.8587 | SRCC = 0.8566 ]
|
| 80 |
+
INFO:root:Average loss=0.14349040169364366
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2601 | LCC = 0.8614 | SRCC = 0.8605 ]
|
| 83 |
+
INFO:root:Average loss=0.13605844850341478
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2414 | LCC = 0.8572 | SRCC = 0.8570 ]
|
| 86 |
+
INFO:root:Average loss=0.1352860598036876
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2451 | LCC = 0.8560 | SRCC = 0.8558 ]
|
| 89 |
+
INFO:root:Average loss=0.13141561834475932
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2441 | LCC = 0.8581 | SRCC = 0.8571 ]
|
| 92 |
+
INFO:root:Average loss=0.1243367326947359
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2485 | LCC = 0.8564 | SRCC = 0.8564 ]
|
| 95 |
+
INFO:root:Average loss=0.12883145486315092
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2562 | LCC = 0.8565 | SRCC = 0.8565 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3199 | LCC = 0.8519 | SRCC = 0.8508 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ca08dd9b5e33413b8dfb4bfeb23a7d2e2aeab58e3e5e42db51e17a3a56114e08
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:37a7c12a4e14f45e409ca9e3dea3ed4b67e48065b2e575c091175d872c1969bc
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.4380430716734666
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3651 | LCC = 0.7911 | SRCC = 0.7934 ]
|
| 8 |
+
INFO:root:Average loss=0.3175906286789821
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3453 | LCC = 0.8137 | SRCC = 0.8168 ]
|
| 11 |
+
INFO:root:Average loss=0.30156727077869266
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3076 | LCC = 0.8227 | SRCC = 0.8246 ]
|
| 14 |
+
INFO:root:Average loss=0.2668433584845983
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3279 | LCC = 0.8348 | SRCC = 0.8328 ]
|
| 17 |
+
INFO:root:Average loss=0.2522684801847507
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3280 | LCC = 0.8371 | SRCC = 0.8353 ]
|
| 20 |
+
INFO:root:Average loss=0.2482475391947306
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3321 | LCC = 0.8457 | SRCC = 0.8466 ]
|
| 23 |
+
INFO:root:Average loss=0.2267345585502111
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2629 | LCC = 0.8489 | SRCC = 0.8479 ]
|
| 26 |
+
INFO:root:Average loss=0.23283594827621412
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2862 | LCC = 0.8523 | SRCC = 0.8527 ]
|
| 29 |
+
INFO:root:Average loss=0.23233816027641296
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3290 | LCC = 0.8566 | SRCC = 0.8573 ]
|
| 32 |
+
INFO:root:Average loss=0.217007499283705
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2853 | LCC = 0.8593 | SRCC = 0.8587 ]
|
| 35 |
+
INFO:root:Average loss=0.21657805335827363
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2816 | LCC = 0.8598 | SRCC = 0.8581 ]
|
| 38 |
+
INFO:root:Average loss=0.21848731402021188
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2686 | LCC = 0.8609 | SRCC = 0.8591 ]
|
| 41 |
+
INFO:root:Average loss=0.2011777832148931
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3169 | LCC = 0.8572 | SRCC = 0.8586 ]
|
| 44 |
+
INFO:root:Average loss=0.18928339083989462
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2622 | LCC = 0.8605 | SRCC = 0.8581 ]
|
| 47 |
+
INFO:root:Average loss=0.19051613677770662
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2824 | LCC = 0.8590 | SRCC = 0.8566 ]
|
| 50 |
+
INFO:root:Average loss=0.18298456187431628
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2672 | LCC = 0.8652 | SRCC = 0.8638 ]
|
| 53 |
+
INFO:root:Average loss=0.17545409223590142
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2493 | LCC = 0.8627 | SRCC = 0.8610 ]
|
| 56 |
+
INFO:root:Average loss=0.1711289882659912
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2479 | LCC = 0.8610 | SRCC = 0.8615 ]
|
| 59 |
+
INFO:root:Average loss=0.1718349771048778
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2877 | LCC = 0.8571 | SRCC = 0.8563 ]
|
| 62 |
+
INFO:root:Average loss=0.16342714390693566
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2460 | LCC = 0.8560 | SRCC = 0.8543 ]
|
| 65 |
+
INFO:root:Average loss=0.17224891751240462
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2750 | LCC = 0.8583 | SRCC = 0.8568 ]
|
| 68 |
+
INFO:root:Average loss=0.1667442812751501
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2911 | LCC = 0.8527 | SRCC = 0.8518 ]
|
| 71 |
+
INFO:root:Average loss=0.15686444708934197
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2725 | LCC = 0.8543 | SRCC = 0.8517 ]
|
| 74 |
+
INFO:root:Average loss=0.14840580427493805
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2502 | LCC = 0.8616 | SRCC = 0.8595 ]
|
| 77 |
+
INFO:root:Average loss=0.15023532681740248
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2550 | LCC = 0.8574 | SRCC = 0.8574 ]
|
| 80 |
+
INFO:root:Average loss=0.13739874433630553
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2442 | LCC = 0.8603 | SRCC = 0.8593 ]
|
| 83 |
+
INFO:root:Average loss=0.13486138492440566
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2634 | LCC = 0.8524 | SRCC = 0.8510 ]
|
| 86 |
+
INFO:root:Average loss=0.13031689574321112
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2684 | LCC = 0.8541 | SRCC = 0.8516 ]
|
| 89 |
+
INFO:root:Average loss=0.13020408506958914
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2748 | LCC = 0.8530 | SRCC = 0.8532 ]
|
| 92 |
+
INFO:root:Average loss=0.13747126618639016
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2592 | LCC = 0.8526 | SRCC = 0.8527 ]
|
| 95 |
+
INFO:root:Average loss=0.12138828176718491
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2682 | LCC = 0.8466 | SRCC = 0.8427 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2706 | LCC = 0.8554 | SRCC = 0.8557 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b548a25e71148bd5546559aae6dd1ece4414ac7f19ff6949e0c017e983bbd436
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:61376ce0e5e7483513dca887e6ef4f2198e93a888ed19f707b1657d6d4aae2f2
|
| 3 |
+
size 2749702
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.5438558818438114
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3763 | LCC = 0.7776 | SRCC = 0.7818 ]
|
| 8 |
+
INFO:root:Average loss=0.31684504716824263
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3293 | LCC = 0.8029 | SRCC = 0.8075 ]
|
| 11 |
+
INFO:root:Average loss=0.29425239715820706
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3285 | LCC = 0.8216 | SRCC = 0.8232 ]
|
| 14 |
+
INFO:root:Average loss=0.2700358713284517
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3121 | LCC = 0.8352 | SRCC = 0.8358 ]
|
| 17 |
+
INFO:root:Average loss=0.26226788205214036
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3116 | LCC = 0.8435 | SRCC = 0.8439 ]
|
| 20 |
+
INFO:root:Average loss=0.25144613820772904
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2722 | LCC = 0.8406 | SRCC = 0.8421 ]
|
| 23 |
+
INFO:root:Average loss=0.24430168592012846
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2788 | LCC = 0.8474 | SRCC = 0.8465 ]
|
| 26 |
+
INFO:root:Average loss=0.23114579151838255
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3113 | LCC = 0.8413 | SRCC = 0.8448 ]
|
| 29 |
+
INFO:root:Average loss=0.22467450759349725
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2778 | LCC = 0.8434 | SRCC = 0.8444 ]
|
| 32 |
+
INFO:root:Average loss=0.23356003715441778
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2907 | LCC = 0.8512 | SRCC = 0.8503 ]
|
| 35 |
+
INFO:root:Average loss=0.21789864087716126
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2630 | LCC = 0.8572 | SRCC = 0.8562 ]
|
| 38 |
+
INFO:root:Average loss=0.2068679734873466
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2605 | LCC = 0.8578 | SRCC = 0.8578 ]
|
| 41 |
+
INFO:root:Average loss=0.2001887424251972
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2562 | LCC = 0.8586 | SRCC = 0.8572 ]
|
| 44 |
+
INFO:root:Average loss=0.1998200210241171
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2862 | LCC = 0.8608 | SRCC = 0.8604 ]
|
| 47 |
+
INFO:root:Average loss=0.1897058802155348
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2510 | LCC = 0.8601 | SRCC = 0.8589 ]
|
| 50 |
+
INFO:root:Average loss=0.18851592544561777
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2587 | LCC = 0.8573 | SRCC = 0.8566 ]
|
| 53 |
+
INFO:root:Average loss=0.1853850024441878
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2748 | LCC = 0.8663 | SRCC = 0.8648 ]
|
| 56 |
+
INFO:root:Average loss=0.17419022197524706
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2541 | LCC = 0.8611 | SRCC = 0.8600 ]
|
| 59 |
+
INFO:root:Average loss=0.17481951577923235
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2906 | LCC = 0.8537 | SRCC = 0.8595 ]
|
| 62 |
+
INFO:root:Average loss=0.16672851966741759
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2557 | LCC = 0.8599 | SRCC = 0.8582 ]
|
| 65 |
+
INFO:root:Average loss=0.16473962548069465
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2713 | LCC = 0.8644 | SRCC = 0.8624 ]
|
| 68 |
+
INFO:root:Average loss=0.1695888197192779
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2281 | LCC = 0.8591 | SRCC = 0.8583 ]
|
| 71 |
+
INFO:root:Average loss=0.15933551925879258
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2526 | LCC = 0.8579 | SRCC = 0.8556 ]
|
| 74 |
+
INFO:root:Average loss=0.14816596855719885
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2568 | LCC = 0.8595 | SRCC = 0.8589 ]
|
| 77 |
+
INFO:root:Average loss=0.15036036523106772
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2616 | LCC = 0.8517 | SRCC = 0.8531 ]
|
| 80 |
+
INFO:root:Average loss=0.1405003054592854
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2301 | LCC = 0.8589 | SRCC = 0.8591 ]
|
| 83 |
+
INFO:root:Average loss=0.140032014499108
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2576 | LCC = 0.8561 | SRCC = 0.8565 ]
|
| 86 |
+
INFO:root:Average loss=0.1408207094631134
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2308 | LCC = 0.8619 | SRCC = 0.8610 ]
|
| 89 |
+
INFO:root:Average loss=0.1366943697898816
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2400 | LCC = 0.8593 | SRCC = 0.8582 ]
|
| 92 |
+
INFO:root:Average loss=0.13014958646052924
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2415 | LCC = 0.8564 | SRCC = 0.8551 ]
|
| 95 |
+
INFO:root:Average loss=0.12681995341793084
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2585 | LCC = 0.8520 | SRCC = 0.8501 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2955 | LCC = 0.8546 | SRCC = 0.8541 ]
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cc97282c85157499e35d31ae8323f0ebf4c60e9b69d16a3f7b9cf28675c87eeb
|
| 3 |
+
size 2747979
|
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e48897cd2e4fd733ede66bbcaa4e114dbc7093a2593f07663faeb8f0f94226ed
|
| 3 |
+
size 2749702
|