Upload SSL Layer MOS checkpoints (part 20)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_tencent/w2v2_xlsr_300m/layer23/run1/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run1/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer23/run2/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer23/run2/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run2/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run2/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer23/run3/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer23/run3/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run3/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run3/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer23/run4/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer23/run4/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run4/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer23/run4/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer3/run0/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer3/run0/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run0/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run0/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer3/run1/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer3/run1/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run1/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run1/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer3/run2/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer3/run2/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run2/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run2/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer3/run3/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer3/run3/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run3/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run3/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer3/run4/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer3/run4/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run4/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer3/run4/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer4/run0/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer4/run0/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run0/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run0/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer4/run1/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer4/run1/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run1/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run1/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer4/run2/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer4/run2/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run2/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run2/train.log +99 -0
- results_tencent/w2v2_xlsr_300m/layer4/run3/config.gin +31 -0
- results_tencent/w2v2_xlsr_300m/layer4/run3/model.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run3/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_300m/layer4/run3/train.log +99 -0
results_tencent/w2v2_xlsr_300m/layer23/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9deed34687949bbdffbdbca2eaf80b31ca9490445ba34ab25b807c663d058f47
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer23/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=1.0563081212043761
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.7386 | LCC = 0.8590 | SRCC = 0.8606 ]
|
| 8 |
+
INFO:root:Average loss=0.43636824917793277
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4267 | LCC = 0.8996 | SRCC = 0.8935 ]
|
| 11 |
+
INFO:root:Average loss=0.3665706579685211
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.5825 | LCC = 0.8866 | SRCC = 0.9005 ]
|
| 14 |
+
INFO:root:Average loss=0.3331493958234787
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3987 | LCC = 0.9092 | SRCC = 0.9098 ]
|
| 17 |
+
INFO:root:Average loss=0.2601349173784256
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3710 | LCC = 0.9199 | SRCC = 0.9199 ]
|
| 20 |
+
INFO:root:Average loss=0.2473576443195343
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3600 | LCC = 0.9305 | SRCC = 0.9290 ]
|
| 23 |
+
INFO:root:Average loss=0.2470013654232025
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2677 | LCC = 0.9316 | SRCC = 0.9299 ]
|
| 26 |
+
INFO:root:Average loss=0.2507935168147087
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2492 | LCC = 0.9306 | SRCC = 0.9307 ]
|
| 29 |
+
INFO:root:Average loss=0.19358640390634538
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2317 | LCC = 0.9388 | SRCC = 0.9382 ]
|
| 32 |
+
INFO:root:Average loss=0.18437114834785462
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2824 | LCC = 0.9446 | SRCC = 0.9433 ]
|
| 35 |
+
INFO:root:Average loss=0.2348708925843239
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2946 | LCC = 0.9454 | SRCC = 0.9464 ]
|
| 38 |
+
INFO:root:Average loss=0.17242506766319274
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2381 | LCC = 0.9443 | SRCC = 0.9461 ]
|
| 41 |
+
INFO:root:Average loss=0.1622229313850403
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2086 | LCC = 0.9495 | SRCC = 0.9496 ]
|
| 44 |
+
INFO:root:Average loss=0.1574323083758354
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2053 | LCC = 0.9508 | SRCC = 0.9516 ]
|
| 47 |
+
INFO:root:Average loss=0.15850027018785476
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3115 | LCC = 0.9474 | SRCC = 0.9513 ]
|
| 50 |
+
INFO:root:Average loss=0.15702237820625306
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2225 | LCC = 0.9545 | SRCC = 0.9541 ]
|
| 53 |
+
INFO:root:Average loss=0.1442300369143486
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1936 | LCC = 0.9506 | SRCC = 0.9516 ]
|
| 56 |
+
INFO:root:Average loss=0.13388900846242904
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2077 | LCC = 0.9542 | SRCC = 0.9550 ]
|
| 59 |
+
INFO:root:Average loss=0.13456004190444945
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1951 | LCC = 0.9569 | SRCC = 0.9564 ]
|
| 62 |
+
INFO:root:Average loss=0.12940751343965531
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1985 | LCC = 0.9515 | SRCC = 0.9543 ]
|
| 65 |
+
INFO:root:Average loss=0.12696247392892837
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2164 | LCC = 0.9575 | SRCC = 0.9577 ]
|
| 68 |
+
INFO:root:Average loss=0.12185577318072319
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2530 | LCC = 0.9567 | SRCC = 0.9574 ]
|
| 71 |
+
INFO:root:Average loss=0.11650395280122756
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2103 | LCC = 0.9586 | SRCC = 0.9585 ]
|
| 74 |
+
INFO:root:Average loss=0.12136834174394608
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2055 | LCC = 0.9589 | SRCC = 0.9581 ]
|
| 77 |
+
INFO:root:Average loss=0.1141653277873993
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2030 | LCC = 0.9571 | SRCC = 0.9574 ]
|
| 80 |
+
INFO:root:Average loss=0.11773455655574798
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2126 | LCC = 0.9608 | SRCC = 0.9602 ]
|
| 83 |
+
INFO:root:Average loss=0.12199297505617142
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2463 | LCC = 0.9562 | SRCC = 0.9571 ]
|
| 86 |
+
INFO:root:Average loss=0.10990665045380592
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1957 | LCC = 0.9615 | SRCC = 0.9606 ]
|
| 89 |
+
INFO:root:Average loss=0.10368599903583527
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2125 | LCC = 0.9611 | SRCC = 0.9612 ]
|
| 92 |
+
INFO:root:Average loss=0.10295946589112281
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2154 | LCC = 0.9609 | SRCC = 0.9608 ]
|
| 95 |
+
INFO:root:Average loss=0.10441702657938004
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2320 | LCC = 0.9566 | SRCC = 0.9582 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2017 | LCC = 0.9608 | SRCC = 0.9615 ]
|
results_tencent/w2v2_xlsr_300m/layer23/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer23/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b593f01a812561d51caac313bc5c6bd12401973794f032e49f13e1f2fe69ed9
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer23/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0e7f331b68e09a285786ae12faee10f37f40d1ec8e825a5af4f258d78e42f59a
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer23/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=1.2454368045330049
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.9272 | LCC = 0.7573 | SRCC = 0.7894 ]
|
| 8 |
+
INFO:root:Average loss=0.4070986392498016
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3697 | LCC = 0.8888 | SRCC = 0.8883 ]
|
| 11 |
+
INFO:root:Average loss=0.3334553905725479
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.5089 | LCC = 0.8980 | SRCC = 0.9068 ]
|
| 14 |
+
INFO:root:Average loss=0.2786888217926025
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3937 | LCC = 0.9223 | SRCC = 0.9196 ]
|
| 17 |
+
INFO:root:Average loss=0.2641449633836746
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3269 | LCC = 0.9274 | SRCC = 0.9255 ]
|
| 20 |
+
INFO:root:Average loss=0.2518278401494026
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2548 | LCC = 0.9168 | SRCC = 0.9214 ]
|
| 23 |
+
INFO:root:Average loss=0.23646544587612153
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2542 | LCC = 0.9330 | SRCC = 0.9331 ]
|
| 26 |
+
INFO:root:Average loss=0.2051917288303375
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2680 | LCC = 0.9354 | SRCC = 0.9349 ]
|
| 29 |
+
INFO:root:Average loss=0.2318065767288208
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2200 | LCC = 0.9394 | SRCC = 0.9400 ]
|
| 32 |
+
INFO:root:Average loss=0.18956960862874986
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3810 | LCC = 0.9370 | SRCC = 0.9452 ]
|
| 35 |
+
INFO:root:Average loss=0.20585258382558821
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2534 | LCC = 0.9421 | SRCC = 0.9434 ]
|
| 38 |
+
INFO:root:Average loss=0.18163397318124772
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2232 | LCC = 0.9471 | SRCC = 0.9473 ]
|
| 41 |
+
INFO:root:Average loss=0.1720084508061409
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2954 | LCC = 0.9487 | SRCC = 0.9500 ]
|
| 44 |
+
INFO:root:Average loss=0.14915228682756423
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2298 | LCC = 0.9519 | SRCC = 0.9517 ]
|
| 47 |
+
INFO:root:Average loss=0.14277833980321883
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2364 | LCC = 0.9460 | SRCC = 0.9457 ]
|
| 50 |
+
INFO:root:Average loss=0.1492177627682686
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2449 | LCC = 0.9534 | SRCC = 0.9544 ]
|
| 53 |
+
INFO:root:Average loss=0.13858777248859405
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2090 | LCC = 0.9544 | SRCC = 0.9542 ]
|
| 56 |
+
INFO:root:Average loss=0.14254310512542726
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2187 | LCC = 0.9552 | SRCC = 0.9554 ]
|
| 59 |
+
INFO:root:Average loss=0.13297844725847244
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1940 | LCC = 0.9534 | SRCC = 0.9547 ]
|
| 62 |
+
INFO:root:Average loss=0.12449122136831284
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2208 | LCC = 0.9564 | SRCC = 0.9567 ]
|
| 65 |
+
INFO:root:Average loss=0.12975050085783005
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2192 | LCC = 0.9539 | SRCC = 0.9550 ]
|
| 68 |
+
INFO:root:Average loss=0.12197551834583283
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1840 | LCC = 0.9570 | SRCC = 0.9582 ]
|
| 71 |
+
INFO:root:Average loss=0.12703324615955353
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1805 | LCC = 0.9524 | SRCC = 0.9542 ]
|
| 74 |
+
INFO:root:Average loss=0.12633594971895218
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2874 | LCC = 0.9577 | SRCC = 0.9599 ]
|
| 77 |
+
INFO:root:Average loss=0.11747282934188842
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1975 | LCC = 0.9587 | SRCC = 0.9593 ]
|
| 80 |
+
INFO:root:Average loss=0.11518598806858063
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1968 | LCC = 0.9603 | SRCC = 0.9604 ]
|
| 83 |
+
INFO:root:Average loss=0.1197339205443859
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2631 | LCC = 0.9585 | SRCC = 0.9606 ]
|
| 86 |
+
INFO:root:Average loss=0.10964244675636292
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1846 | LCC = 0.9610 | SRCC = 0.9612 ]
|
| 89 |
+
INFO:root:Average loss=0.11100984755158425
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2261 | LCC = 0.9599 | SRCC = 0.9605 ]
|
| 92 |
+
INFO:root:Average loss=0.10690061169862747
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1774 | LCC = 0.9595 | SRCC = 0.9594 ]
|
| 95 |
+
INFO:root:Average loss=0.1042147558927536
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2075 | LCC = 0.9612 | SRCC = 0.9615 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2153 | LCC = 0.9599 | SRCC = 0.9620 ]
|
results_tencent/w2v2_xlsr_300m/layer23/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer23/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bb453cf8d22c715acdbfc7a44d78c26f4604b6ef363206f5c7b9125be580437a
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer23/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b2e395f1a1a88a57c0b9762c97f53edde8094ad76a76e45587acef062a39651b
|
| 3 |
+
size 3404998
|
results_tencent/w2v2_xlsr_300m/layer23/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=1.0217993781566619
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 1.0591 | LCC = 0.8105 | SRCC = 0.8568 ]
|
| 8 |
+
INFO:root:Average loss=0.4108892401456833
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3454 | LCC = 0.8973 | SRCC = 0.8937 ]
|
| 11 |
+
INFO:root:Average loss=0.2908277262449265
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.4217 | LCC = 0.9067 | SRCC = 0.9095 ]
|
| 14 |
+
INFO:root:Average loss=0.3161389358043671
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.4445 | LCC = 0.9103 | SRCC = 0.9153 ]
|
| 17 |
+
INFO:root:Average loss=0.30991377222537997
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3158 | LCC = 0.9249 | SRCC = 0.9241 ]
|
| 20 |
+
INFO:root:Average loss=0.255293332695961
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2536 | LCC = 0.9333 | SRCC = 0.9319 ]
|
| 23 |
+
INFO:root:Average loss=0.21495477044582367
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2286 | LCC = 0.9373 | SRCC = 0.9361 ]
|
| 26 |
+
INFO:root:Average loss=0.19873052734136581
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3152 | LCC = 0.9417 | SRCC = 0.9403 ]
|
| 29 |
+
INFO:root:Average loss=0.18445929199457167
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2543 | LCC = 0.9417 | SRCC = 0.9419 ]
|
| 32 |
+
INFO:root:Average loss=0.18127670788764955
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3192 | LCC = 0.9437 | SRCC = 0.9436 ]
|
| 35 |
+
INFO:root:Average loss=0.1928513303399086
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2450 | LCC = 0.9396 | SRCC = 0.9439 ]
|
| 38 |
+
INFO:root:Average loss=0.16134409826993942
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2407 | LCC = 0.9483 | SRCC = 0.9472 ]
|
| 41 |
+
INFO:root:Average loss=0.17079754358530044
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2452 | LCC = 0.9476 | SRCC = 0.9474 ]
|
| 44 |
+
INFO:root:Average loss=0.15236402332782745
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2353 | LCC = 0.9485 | SRCC = 0.9489 ]
|
| 47 |
+
INFO:root:Average loss=0.14256081676483154
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1954 | LCC = 0.9518 | SRCC = 0.9512 ]
|
| 50 |
+
INFO:root:Average loss=0.1471487997174263
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1737 | LCC = 0.9513 | SRCC = 0.9529 ]
|
| 53 |
+
INFO:root:Average loss=0.1566015036702156
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2357 | LCC = 0.9533 | SRCC = 0.9523 ]
|
| 56 |
+
INFO:root:Average loss=0.13364748448133468
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2799 | LCC = 0.9519 | SRCC = 0.9543 ]
|
| 59 |
+
INFO:root:Average loss=0.15910709178447724
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2889 | LCC = 0.9459 | SRCC = 0.9503 ]
|
| 62 |
+
INFO:root:Average loss=0.14192144161462783
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2777 | LCC = 0.9451 | SRCC = 0.9495 ]
|
| 65 |
+
INFO:root:Average loss=0.13185440748929977
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2025 | LCC = 0.9559 | SRCC = 0.9554 ]
|
| 68 |
+
INFO:root:Average loss=0.12572974821925162
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2025 | LCC = 0.9531 | SRCC = 0.9543 ]
|
| 71 |
+
INFO:root:Average loss=0.12733065366744994
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2168 | LCC = 0.9511 | SRCC = 0.9532 ]
|
| 74 |
+
INFO:root:Average loss=0.11917251497507095
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2036 | LCC = 0.9563 | SRCC = 0.9554 ]
|
| 77 |
+
INFO:root:Average loss=0.11531247371435166
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2819 | LCC = 0.9509 | SRCC = 0.9522 ]
|
| 80 |
+
INFO:root:Average loss=0.12601213324069976
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2440 | LCC = 0.9578 | SRCC = 0.9572 ]
|
| 83 |
+
INFO:root:Average loss=0.12161339432001114
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2255 | LCC = 0.9585 | SRCC = 0.9577 ]
|
| 86 |
+
INFO:root:Average loss=0.1213982959985733
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2173 | LCC = 0.9592 | SRCC = 0.9581 ]
|
| 89 |
+
INFO:root:Average loss=0.11161489087343215
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1747 | LCC = 0.9583 | SRCC = 0.9578 ]
|
| 92 |
+
INFO:root:Average loss=0.11380220684409141
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2964 | LCC = 0.9556 | SRCC = 0.9580 ]
|
| 95 |
+
INFO:root:Average loss=0.10564400154352188
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2176 | LCC = 0.9596 | SRCC = 0.9592 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2175 | LCC = 0.9600 | SRCC = 0.9606 ]
|
results_tencent/w2v2_xlsr_300m/layer23/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer23/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3a66760664507800bd9cd629fc085c12ea15f1ae5d31c4e062684591f1cfb71c
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer23/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c3e0be75784aed4989877cfd11a53e63f20cfeb72c41b1bcc4a6e4dfc44b2f8f
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer23/run4/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.8145056846141815
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5079 | LCC = 0.8813 | SRCC = 0.8778 ]
|
| 8 |
+
INFO:root:Average loss=0.3912081800699234
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4546 | LCC = 0.8927 | SRCC = 0.8958 ]
|
| 11 |
+
INFO:root:Average loss=0.3569305625557899
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3900 | LCC = 0.9121 | SRCC = 0.9098 ]
|
| 14 |
+
INFO:root:Average loss=0.37009758031368256
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2886 | LCC = 0.9206 | SRCC = 0.9167 ]
|
| 17 |
+
INFO:root:Average loss=0.2958523344993591
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.4682 | LCC = 0.9135 | SRCC = 0.9183 ]
|
| 20 |
+
INFO:root:Average loss=0.22553248322010042
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.5679 | LCC = 0.9123 | SRCC = 0.9226 ]
|
| 23 |
+
INFO:root:Average loss=0.2567094199061394
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3188 | LCC = 0.9405 | SRCC = 0.9380 ]
|
| 26 |
+
INFO:root:Average loss=0.22448823195695877
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3060 | LCC = 0.9410 | SRCC = 0.9395 ]
|
| 29 |
+
INFO:root:Average loss=0.1885664335489273
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2902 | LCC = 0.9383 | SRCC = 0.9398 ]
|
| 32 |
+
INFO:root:Average loss=0.19456501942873
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2486 | LCC = 0.9459 | SRCC = 0.9457 ]
|
| 35 |
+
INFO:root:Average loss=0.17806837284564972
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2564 | LCC = 0.9453 | SRCC = 0.9455 ]
|
| 38 |
+
INFO:root:Average loss=0.15499526584148407
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2147 | LCC = 0.9500 | SRCC = 0.9502 ]
|
| 41 |
+
INFO:root:Average loss=0.16285975557565688
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3589 | LCC = 0.9488 | SRCC = 0.9494 ]
|
| 44 |
+
INFO:root:Average loss=0.16125369787216187
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2220 | LCC = 0.9510 | SRCC = 0.9511 ]
|
| 47 |
+
INFO:root:Average loss=0.15557553827762605
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2578 | LCC = 0.9507 | SRCC = 0.9515 ]
|
| 50 |
+
INFO:root:Average loss=0.1599581662416458
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2192 | LCC = 0.9520 | SRCC = 0.9525 ]
|
| 53 |
+
INFO:root:Average loss=0.13801892495155335
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2273 | LCC = 0.9542 | SRCC = 0.9537 ]
|
| 56 |
+
INFO:root:Average loss=0.13788513362407684
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2258 | LCC = 0.9538 | SRCC = 0.9534 ]
|
| 59 |
+
INFO:root:Average loss=0.1377027438879013
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2296 | LCC = 0.9512 | SRCC = 0.9506 ]
|
| 62 |
+
INFO:root:Average loss=0.12967479968070983
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2255 | LCC = 0.9553 | SRCC = 0.9554 ]
|
| 65 |
+
INFO:root:Average loss=0.1240396209359169
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2883 | LCC = 0.9521 | SRCC = 0.9537 ]
|
| 68 |
+
INFO:root:Average loss=0.12480794876813889
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2768 | LCC = 0.9582 | SRCC = 0.9584 ]
|
| 71 |
+
INFO:root:Average loss=0.12560943567752839
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2291 | LCC = 0.9550 | SRCC = 0.9553 ]
|
| 74 |
+
INFO:root:Average loss=0.1200638672709465
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2564 | LCC = 0.9553 | SRCC = 0.9568 ]
|
| 77 |
+
INFO:root:Average loss=0.13097032684087753
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2212 | LCC = 0.9524 | SRCC = 0.9533 ]
|
| 80 |
+
INFO:root:Average loss=0.12304448008537293
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2513 | LCC = 0.9527 | SRCC = 0.9550 ]
|
| 83 |
+
INFO:root:Average loss=0.11969953566789628
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2351 | LCC = 0.9596 | SRCC = 0.9595 ]
|
| 86 |
+
INFO:root:Average loss=0.11048496359586715
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2014 | LCC = 0.9593 | SRCC = 0.9589 ]
|
| 89 |
+
INFO:root:Average loss=0.11392447927594185
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2106 | LCC = 0.9587 | SRCC = 0.9597 ]
|
| 92 |
+
INFO:root:Average loss=0.10840985369682311
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2144 | LCC = 0.9603 | SRCC = 0.9598 ]
|
| 95 |
+
INFO:root:Average loss=0.1037284859418869
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2233 | LCC = 0.9595 | SRCC = 0.9591 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2146 | LCC = 0.9598 | SRCC = 0.9605 ]
|
results_tencent/w2v2_xlsr_300m/layer3/run0/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer3/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7d4d14ae671b197d8d8d4109c100984582b288e047ee8c08c6b977cdcc74ace9
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer3/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bd84e031d1e5a2a42d589559e6fad5afc96763ed128f23626e2a985e9aa5e1db
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer3/run0/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.4928909802436829
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2999 | LCC = 0.9365 | SRCC = 0.9372 ]
|
| 8 |
+
INFO:root:Average loss=0.16937899965047837
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2725 | LCC = 0.9498 | SRCC = 0.9495 ]
|
| 11 |
+
INFO:root:Average loss=0.13991226106882096
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2283 | LCC = 0.9555 | SRCC = 0.9554 ]
|
| 14 |
+
INFO:root:Average loss=0.1192230771780014
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2289 | LCC = 0.9601 | SRCC = 0.9603 ]
|
| 17 |
+
INFO:root:Average loss=0.1098528333902359
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2098 | LCC = 0.9632 | SRCC = 0.9627 ]
|
| 20 |
+
INFO:root:Average loss=0.09928976985812187
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1914 | LCC = 0.9636 | SRCC = 0.9632 ]
|
| 23 |
+
INFO:root:Average loss=0.08686692520976066
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1928 | LCC = 0.9659 | SRCC = 0.9655 ]
|
| 26 |
+
INFO:root:Average loss=0.08223827317357063
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2097 | LCC = 0.9663 | SRCC = 0.9661 ]
|
| 29 |
+
INFO:root:Average loss=0.07617785930633544
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2168 | LCC = 0.9658 | SRCC = 0.9655 ]
|
| 32 |
+
INFO:root:Average loss=0.07284880968928337
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1634 | LCC = 0.9659 | SRCC = 0.9659 ]
|
| 35 |
+
INFO:root:Average loss=0.0717342868745327
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2133 | LCC = 0.9675 | SRCC = 0.9672 ]
|
| 38 |
+
INFO:root:Average loss=0.0682714629471302
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2210 | LCC = 0.9684 | SRCC = 0.9691 ]
|
| 41 |
+
INFO:root:Average loss=0.06349933075904846
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2104 | LCC = 0.9688 | SRCC = 0.9687 ]
|
| 44 |
+
INFO:root:Average loss=0.06211875455081463
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1813 | LCC = 0.9678 | SRCC = 0.9668 ]
|
| 47 |
+
INFO:root:Average loss=0.057986576482653616
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1774 | LCC = 0.9704 | SRCC = 0.9691 ]
|
| 50 |
+
INFO:root:Average loss=0.05545867249369621
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1887 | LCC = 0.9707 | SRCC = 0.9698 ]
|
| 53 |
+
INFO:root:Average loss=0.05239409136772156
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2099 | LCC = 0.9685 | SRCC = 0.9690 ]
|
| 56 |
+
INFO:root:Average loss=0.04977992175519466
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1832 | LCC = 0.9700 | SRCC = 0.9689 ]
|
| 59 |
+
INFO:root:Average loss=0.046591076463460925
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1702 | LCC = 0.9707 | SRCC = 0.9696 ]
|
| 62 |
+
INFO:root:Average loss=0.04589277376234532
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1565 | LCC = 0.9706 | SRCC = 0.9693 ]
|
| 65 |
+
INFO:root:Average loss=0.04347516365349293
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1732 | LCC = 0.9714 | SRCC = 0.9702 ]
|
| 68 |
+
INFO:root:Average loss=0.043134723335504534
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2012 | LCC = 0.9705 | SRCC = 0.9698 ]
|
| 71 |
+
INFO:root:Average loss=0.043562618657946584
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1732 | LCC = 0.9713 | SRCC = 0.9704 ]
|
| 74 |
+
INFO:root:Average loss=0.03857028645277023
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1833 | LCC = 0.9714 | SRCC = 0.9702 ]
|
| 77 |
+
INFO:root:Average loss=0.03828483219444752
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1784 | LCC = 0.9701 | SRCC = 0.9692 ]
|
| 80 |
+
INFO:root:Average loss=0.03631746079027653
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1682 | LCC = 0.9707 | SRCC = 0.9698 ]
|
| 83 |
+
INFO:root:Average loss=0.03477946110069752
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1900 | LCC = 0.9716 | SRCC = 0.9703 ]
|
| 86 |
+
INFO:root:Average loss=0.0341524433195591
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1786 | LCC = 0.9721 | SRCC = 0.9710 ]
|
| 89 |
+
INFO:root:Average loss=0.034333244621753696
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1568 | LCC = 0.9712 | SRCC = 0.9698 ]
|
| 92 |
+
INFO:root:Average loss=0.03195479291677475
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1698 | LCC = 0.9702 | SRCC = 0.9688 ]
|
| 95 |
+
INFO:root:Average loss=0.031488272041082385
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1602 | LCC = 0.9714 | SRCC = 0.9706 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1861 | LCC = 0.9690 | SRCC = 0.9692 ]
|
results_tencent/w2v2_xlsr_300m/layer3/run1/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer3/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06f970ccbfeeeda29451662d84976e2f04a8b09cb3b2810cb82e2e6e76166755
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer3/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:26498e14c298d2fbb45f2bc25c7a245339cba729de222da271f483f6695bf26f
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer3/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.39172349286079405
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2452 | LCC = 0.9335 | SRCC = 0.9357 ]
|
| 8 |
+
INFO:root:Average loss=0.1631543598175049
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2577 | LCC = 0.9484 | SRCC = 0.9463 ]
|
| 11 |
+
INFO:root:Average loss=0.1373210073709488
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2732 | LCC = 0.9541 | SRCC = 0.9538 ]
|
| 14 |
+
INFO:root:Average loss=0.11968734902143478
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2440 | LCC = 0.9566 | SRCC = 0.9559 ]
|
| 17 |
+
INFO:root:Average loss=0.10614920368790627
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2279 | LCC = 0.9601 | SRCC = 0.9598 ]
|
| 20 |
+
INFO:root:Average loss=0.09772597977519035
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2263 | LCC = 0.9620 | SRCC = 0.9615 ]
|
| 23 |
+
INFO:root:Average loss=0.08985240116715432
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1695 | LCC = 0.9616 | SRCC = 0.9620 ]
|
| 26 |
+
INFO:root:Average loss=0.08811028820276261
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1973 | LCC = 0.9642 | SRCC = 0.9635 ]
|
| 29 |
+
INFO:root:Average loss=0.07966661924123764
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1957 | LCC = 0.9650 | SRCC = 0.9642 ]
|
| 32 |
+
INFO:root:Average loss=0.07141376048326492
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2288 | LCC = 0.9645 | SRCC = 0.9640 ]
|
| 35 |
+
INFO:root:Average loss=0.0724522342979908
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2128 | LCC = 0.9648 | SRCC = 0.9642 ]
|
| 38 |
+
INFO:root:Average loss=0.06668364444375038
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1947 | LCC = 0.9634 | SRCC = 0.9622 ]
|
| 41 |
+
INFO:root:Average loss=0.06506802225112915
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2270 | LCC = 0.9649 | SRCC = 0.9657 ]
|
| 44 |
+
INFO:root:Average loss=0.06406673233211041
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1766 | LCC = 0.9649 | SRCC = 0.9640 ]
|
| 47 |
+
INFO:root:Average loss=0.05752970385551453
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2125 | LCC = 0.9666 | SRCC = 0.9662 ]
|
| 50 |
+
INFO:root:Average loss=0.05704319806396961
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1692 | LCC = 0.9679 | SRCC = 0.9671 ]
|
| 53 |
+
INFO:root:Average loss=0.054945798367261885
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1877 | LCC = 0.9676 | SRCC = 0.9667 ]
|
| 56 |
+
INFO:root:Average loss=0.05074971768260002
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1854 | LCC = 0.9671 | SRCC = 0.9669 ]
|
| 59 |
+
INFO:root:Average loss=0.05246031494438648
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1725 | LCC = 0.9685 | SRCC = 0.9674 ]
|
| 62 |
+
INFO:root:Average loss=0.04877910551428795
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1944 | LCC = 0.9677 | SRCC = 0.9671 ]
|
| 65 |
+
INFO:root:Average loss=0.04537329433858395
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1706 | LCC = 0.9662 | SRCC = 0.9658 ]
|
| 68 |
+
INFO:root:Average loss=0.04382076473534107
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1881 | LCC = 0.9682 | SRCC = 0.9674 ]
|
| 71 |
+
INFO:root:Average loss=0.04116334687173367
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1758 | LCC = 0.9688 | SRCC = 0.9676 ]
|
| 74 |
+
INFO:root:Average loss=0.04096766270697117
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2053 | LCC = 0.9677 | SRCC = 0.9676 ]
|
| 77 |
+
INFO:root:Average loss=0.040873097628355025
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2228 | LCC = 0.9660 | SRCC = 0.9669 ]
|
| 80 |
+
INFO:root:Average loss=0.03826784770190716
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2157 | LCC = 0.9681 | SRCC = 0.9679 ]
|
| 83 |
+
INFO:root:Average loss=0.03710025091469288
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1836 | LCC = 0.9699 | SRCC = 0.9686 ]
|
| 86 |
+
INFO:root:Average loss=0.033851901218295094
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1744 | LCC = 0.9698 | SRCC = 0.9679 ]
|
| 89 |
+
INFO:root:Average loss=0.03558759294450283
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1923 | LCC = 0.9677 | SRCC = 0.9667 ]
|
| 92 |
+
INFO:root:Average loss=0.03433390028774738
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1760 | LCC = 0.9694 | SRCC = 0.9682 ]
|
| 95 |
+
INFO:root:Average loss=0.03310309068858624
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1745 | LCC = 0.9694 | SRCC = 0.9676 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1879 | LCC = 0.9685 | SRCC = 0.9683 ]
|
results_tencent/w2v2_xlsr_300m/layer3/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer3/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8bc80f0dcac434ef2ad23e8b15029f942a75a7659999b36bcf19b35129d420c
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer3/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4a5ef4e1cf484f54a66add24d67127da52aeb8609558fbb7c01a7bc25c2a8f17
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer3/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.32500236386060716
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2885 | LCC = 0.9391 | SRCC = 0.9385 ]
|
| 8 |
+
INFO:root:Average loss=0.1558499710559845
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2587 | LCC = 0.9454 | SRCC = 0.9486 ]
|
| 11 |
+
INFO:root:Average loss=0.14717653560638427
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2308 | LCC = 0.9543 | SRCC = 0.9554 ]
|
| 14 |
+
INFO:root:Average loss=0.11835578638315201
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2476 | LCC = 0.9586 | SRCC = 0.9592 ]
|
| 17 |
+
INFO:root:Average loss=0.10746299874782562
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2186 | LCC = 0.9583 | SRCC = 0.9602 ]
|
| 20 |
+
INFO:root:Average loss=0.09926072120666504
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1998 | LCC = 0.9634 | SRCC = 0.9630 ]
|
| 23 |
+
INFO:root:Average loss=0.08872406920790672
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1973 | LCC = 0.9637 | SRCC = 0.9632 ]
|
| 26 |
+
INFO:root:Average loss=0.08390092438459397
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1869 | LCC = 0.9633 | SRCC = 0.9646 ]
|
| 29 |
+
INFO:root:Average loss=0.07669991993904114
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2116 | LCC = 0.9656 | SRCC = 0.9659 ]
|
| 32 |
+
INFO:root:Average loss=0.07126889047026634
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1746 | LCC = 0.9659 | SRCC = 0.9649 ]
|
| 35 |
+
INFO:root:Average loss=0.06808758071064949
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1890 | LCC = 0.9682 | SRCC = 0.9678 ]
|
| 38 |
+
INFO:root:Average loss=0.07008291900157929
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1670 | LCC = 0.9680 | SRCC = 0.9672 ]
|
| 41 |
+
INFO:root:Average loss=0.06043029496073723
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2078 | LCC = 0.9666 | SRCC = 0.9666 ]
|
| 44 |
+
INFO:root:Average loss=0.059582085132598876
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2207 | LCC = 0.9656 | SRCC = 0.9672 ]
|
| 47 |
+
INFO:root:Average loss=0.06040261626243591
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1618 | LCC = 0.9674 | SRCC = 0.9673 ]
|
| 50 |
+
INFO:root:Average loss=0.05360208167135715
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1690 | LCC = 0.9683 | SRCC = 0.9672 ]
|
| 53 |
+
INFO:root:Average loss=0.052096041172742844
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1796 | LCC = 0.9685 | SRCC = 0.9681 ]
|
| 56 |
+
INFO:root:Average loss=0.048393024265766145
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1609 | LCC = 0.9681 | SRCC = 0.9675 ]
|
| 59 |
+
INFO:root:Average loss=0.04572081921994686
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1854 | LCC = 0.9679 | SRCC = 0.9672 ]
|
| 62 |
+
INFO:root:Average loss=0.04716121120750904
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1496 | LCC = 0.9687 | SRCC = 0.9676 ]
|
| 65 |
+
INFO:root:Average loss=0.041934609800577165
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1696 | LCC = 0.9680 | SRCC = 0.9676 ]
|
| 68 |
+
INFO:root:Average loss=0.040934306874871255
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1456 | LCC = 0.9690 | SRCC = 0.9683 ]
|
| 71 |
+
INFO:root:Average loss=0.03817667020857334
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1534 | LCC = 0.9689 | SRCC = 0.9686 ]
|
| 74 |
+
INFO:root:Average loss=0.03709316331148148
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1706 | LCC = 0.9693 | SRCC = 0.9683 ]
|
| 77 |
+
INFO:root:Average loss=0.03549412325024605
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1677 | LCC = 0.9696 | SRCC = 0.9687 ]
|
| 80 |
+
INFO:root:Average loss=0.03586291381716728
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1550 | LCC = 0.9690 | SRCC = 0.9686 ]
|
| 83 |
+
INFO:root:Average loss=0.03558068524301052
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1646 | LCC = 0.9688 | SRCC = 0.9679 ]
|
| 86 |
+
INFO:root:Average loss=0.03305184856057167
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1894 | LCC = 0.9691 | SRCC = 0.9689 ]
|
| 89 |
+
INFO:root:Average loss=0.03446996255218983
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1821 | LCC = 0.9677 | SRCC = 0.9676 ]
|
| 92 |
+
INFO:root:Average loss=0.032194572389125826
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2019 | LCC = 0.9685 | SRCC = 0.9686 ]
|
| 95 |
+
INFO:root:Average loss=0.030766275979578494
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1831 | LCC = 0.9691 | SRCC = 0.9688 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1712 | LCC = 0.9675 | SRCC = 0.9672 ]
|
results_tencent/w2v2_xlsr_300m/layer3/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer3/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4cafcac12c56a0906e74f8f43cdf7d6286d09c61c69db21f4a884bce1ba3dd44
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer3/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2c2e119336f1e19c811374bd91b6bba64fd605663728098526b86877a34c93e5
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer3/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.4237779357433319
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2507 | LCC = 0.9407 | SRCC = 0.9412 ]
|
| 8 |
+
INFO:root:Average loss=0.15904512071609497
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.1991 | LCC = 0.9490 | SRCC = 0.9508 ]
|
| 11 |
+
INFO:root:Average loss=0.12917388647794723
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2041 | LCC = 0.9577 | SRCC = 0.9569 ]
|
| 14 |
+
INFO:root:Average loss=0.11946277451515198
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2158 | LCC = 0.9588 | SRCC = 0.9586 ]
|
| 17 |
+
INFO:root:Average loss=0.10236752167344093
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2171 | LCC = 0.9620 | SRCC = 0.9612 ]
|
| 20 |
+
INFO:root:Average loss=0.09482676592469215
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1740 | LCC = 0.9630 | SRCC = 0.9627 ]
|
| 23 |
+
INFO:root:Average loss=0.09287394666671753
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2303 | LCC = 0.9630 | SRCC = 0.9636 ]
|
| 26 |
+
INFO:root:Average loss=0.080492874532938
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1874 | LCC = 0.9646 | SRCC = 0.9643 ]
|
| 29 |
+
INFO:root:Average loss=0.07866597092151642
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2078 | LCC = 0.9654 | SRCC = 0.9661 ]
|
| 32 |
+
INFO:root:Average loss=0.0753565459549427
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1660 | LCC = 0.9678 | SRCC = 0.9673 ]
|
| 35 |
+
INFO:root:Average loss=0.06905877244472504
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2352 | LCC = 0.9674 | SRCC = 0.9670 ]
|
| 38 |
+
INFO:root:Average loss=0.06652959322929382
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1574 | LCC = 0.9678 | SRCC = 0.9670 ]
|
| 41 |
+
INFO:root:Average loss=0.06260098233819007
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1575 | LCC = 0.9675 | SRCC = 0.9663 ]
|
| 44 |
+
INFO:root:Average loss=0.0616500555574894
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1422 | LCC = 0.9684 | SRCC = 0.9682 ]
|
| 47 |
+
INFO:root:Average loss=0.05589770913124084
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1801 | LCC = 0.9684 | SRCC = 0.9676 ]
|
| 50 |
+
INFO:root:Average loss=0.05502425518631935
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1770 | LCC = 0.9687 | SRCC = 0.9681 ]
|
| 53 |
+
INFO:root:Average loss=0.053068888947367665
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1768 | LCC = 0.9675 | SRCC = 0.9669 ]
|
| 56 |
+
INFO:root:Average loss=0.047760760962963104
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1555 | LCC = 0.9669 | SRCC = 0.9675 ]
|
| 59 |
+
INFO:root:Average loss=0.04732796381413937
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1410 | LCC = 0.9684 | SRCC = 0.9677 ]
|
| 62 |
+
INFO:root:Average loss=0.046957701101899146
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1656 | LCC = 0.9698 | SRCC = 0.9687 ]
|
| 65 |
+
INFO:root:Average loss=0.04378133215010166
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1886 | LCC = 0.9693 | SRCC = 0.9685 ]
|
| 68 |
+
INFO:root:Average loss=0.0433526478856802
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1493 | LCC = 0.9690 | SRCC = 0.9679 ]
|
| 71 |
+
INFO:root:Average loss=0.040885006234049795
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1614 | LCC = 0.9694 | SRCC = 0.9679 ]
|
| 74 |
+
INFO:root:Average loss=0.03903745400905609
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1486 | LCC = 0.9690 | SRCC = 0.9685 ]
|
| 77 |
+
INFO:root:Average loss=0.03657390807569027
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1741 | LCC = 0.9701 | SRCC = 0.9688 ]
|
| 80 |
+
INFO:root:Average loss=0.039520827695727345
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1467 | LCC = 0.9693 | SRCC = 0.9679 ]
|
| 83 |
+
INFO:root:Average loss=0.034794993579387666
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1615 | LCC = 0.9705 | SRCC = 0.9692 ]
|
| 86 |
+
INFO:root:Average loss=0.033548924967646596
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1373 | LCC = 0.9692 | SRCC = 0.9691 ]
|
| 89 |
+
INFO:root:Average loss=0.03349556158483028
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1646 | LCC = 0.9684 | SRCC = 0.9678 ]
|
| 92 |
+
INFO:root:Average loss=0.033362504348158836
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1690 | LCC = 0.9712 | SRCC = 0.9702 ]
|
| 95 |
+
INFO:root:Average loss=0.03324629755318165
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1324 | LCC = 0.9701 | SRCC = 0.9700 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1724 | LCC = 0.9692 | SRCC = 0.9691 ]
|
results_tencent/w2v2_xlsr_300m/layer3/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer3/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cc90c8e13e29b36108ea187c896e9630c05a14eb2d56e0b8d3f21926dc39901a
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer3/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf1e9fb85fbb2be8d46594a6cbbe6859145b7660baf40bd32a662ff7d9c850e7
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer3/run4/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.48466843366622925
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2569 | LCC = 0.9384 | SRCC = 0.9373 ]
|
| 8 |
+
INFO:root:Average loss=0.1775839799642563
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2046 | LCC = 0.9454 | SRCC = 0.9473 ]
|
| 11 |
+
INFO:root:Average loss=0.15533198940753937
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1944 | LCC = 0.9498 | SRCC = 0.9514 ]
|
| 14 |
+
INFO:root:Average loss=0.1221833969950676
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2344 | LCC = 0.9492 | SRCC = 0.9519 ]
|
| 17 |
+
INFO:root:Average loss=0.10968787693977355
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2175 | LCC = 0.9615 | SRCC = 0.9612 ]
|
| 20 |
+
INFO:root:Average loss=0.101326884329319
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1744 | LCC = 0.9627 | SRCC = 0.9621 ]
|
| 23 |
+
INFO:root:Average loss=0.09600243681669235
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2125 | LCC = 0.9635 | SRCC = 0.9636 ]
|
| 26 |
+
INFO:root:Average loss=0.08750571379065514
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1742 | LCC = 0.9622 | SRCC = 0.9645 ]
|
| 29 |
+
INFO:root:Average loss=0.07722979310154915
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1612 | LCC = 0.9616 | SRCC = 0.9618 ]
|
| 32 |
+
INFO:root:Average loss=0.07694434255361557
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1627 | LCC = 0.9646 | SRCC = 0.9658 ]
|
| 35 |
+
INFO:root:Average loss=0.07097282034158707
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1605 | LCC = 0.9661 | SRCC = 0.9662 ]
|
| 38 |
+
INFO:root:Average loss=0.0660415803194046
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1641 | LCC = 0.9675 | SRCC = 0.9671 ]
|
| 41 |
+
INFO:root:Average loss=0.0639457170367241
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1561 | LCC = 0.9655 | SRCC = 0.9669 ]
|
| 44 |
+
INFO:root:Average loss=0.06160608339309692
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1573 | LCC = 0.9687 | SRCC = 0.9678 ]
|
| 47 |
+
INFO:root:Average loss=0.05722728738188743
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1853 | LCC = 0.9683 | SRCC = 0.9671 ]
|
| 50 |
+
INFO:root:Average loss=0.056382120430469515
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1571 | LCC = 0.9669 | SRCC = 0.9681 ]
|
| 53 |
+
INFO:root:Average loss=0.052327201813459394
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1402 | LCC = 0.9684 | SRCC = 0.9677 ]
|
| 56 |
+
INFO:root:Average loss=0.050134144946932795
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1659 | LCC = 0.9675 | SRCC = 0.9683 ]
|
| 59 |
+
INFO:root:Average loss=0.04756868174672127
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1527 | LCC = 0.9689 | SRCC = 0.9685 ]
|
| 62 |
+
INFO:root:Average loss=0.04681389385461807
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1563 | LCC = 0.9690 | SRCC = 0.9682 ]
|
| 65 |
+
INFO:root:Average loss=0.044286971300840375
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1652 | LCC = 0.9688 | SRCC = 0.9680 ]
|
| 68 |
+
INFO:root:Average loss=0.04239480862021446
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1540 | LCC = 0.9690 | SRCC = 0.9683 ]
|
| 71 |
+
INFO:root:Average loss=0.04106458252668381
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1500 | LCC = 0.9682 | SRCC = 0.9682 ]
|
| 74 |
+
INFO:root:Average loss=0.040112100258469584
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1790 | LCC = 0.9659 | SRCC = 0.9662 ]
|
| 77 |
+
INFO:root:Average loss=0.03831818917393685
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1645 | LCC = 0.9691 | SRCC = 0.9681 ]
|
| 80 |
+
INFO:root:Average loss=0.03647170528769493
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1448 | LCC = 0.9681 | SRCC = 0.9677 ]
|
| 83 |
+
INFO:root:Average loss=0.03561024002730846
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1561 | LCC = 0.9694 | SRCC = 0.9684 ]
|
| 86 |
+
INFO:root:Average loss=0.035114300280809405
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1830 | LCC = 0.9688 | SRCC = 0.9684 ]
|
| 89 |
+
INFO:root:Average loss=0.03342070460319519
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1677 | LCC = 0.9686 | SRCC = 0.9676 ]
|
| 92 |
+
INFO:root:Average loss=0.032618674844503405
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1756 | LCC = 0.9692 | SRCC = 0.9681 ]
|
| 95 |
+
INFO:root:Average loss=0.03046013058722019
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1590 | LCC = 0.9696 | SRCC = 0.9688 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1594 | LCC = 0.9686 | SRCC = 0.9681 ]
|
results_tencent/w2v2_xlsr_300m/layer4/run0/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer4/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b131ec31001310df34289a8edb00d57a85b28f1284d59dead3ba3805a72e01c
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer4/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:78f13040df8e49be355170ba7e86fe6473da9a7c2f4e585862f13a3cc6387d18
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer4/run0/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.40181518894433976
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2499 | LCC = 0.9466 | SRCC = 0.9464 ]
|
| 8 |
+
INFO:root:Average loss=0.15385954695940018
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2499 | LCC = 0.9537 | SRCC = 0.9529 ]
|
| 11 |
+
INFO:root:Average loss=0.12373063308000565
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2109 | LCC = 0.9604 | SRCC = 0.9592 ]
|
| 14 |
+
INFO:root:Average loss=0.10653454419970512
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2147 | LCC = 0.9632 | SRCC = 0.9622 ]
|
| 17 |
+
INFO:root:Average loss=0.09398544239997864
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2260 | LCC = 0.9643 | SRCC = 0.9635 ]
|
| 20 |
+
INFO:root:Average loss=0.0881934364438057
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1943 | LCC = 0.9658 | SRCC = 0.9660 ]
|
| 23 |
+
INFO:root:Average loss=0.07958787786960601
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1783 | LCC = 0.9675 | SRCC = 0.9673 ]
|
| 26 |
+
INFO:root:Average loss=0.0724699057340622
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1683 | LCC = 0.9653 | SRCC = 0.9650 ]
|
| 29 |
+
INFO:root:Average loss=0.06957830360531807
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1751 | LCC = 0.9687 | SRCC = 0.9681 ]
|
| 32 |
+
INFO:root:Average loss=0.06896216365695
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1896 | LCC = 0.9698 | SRCC = 0.9689 ]
|
| 35 |
+
INFO:root:Average loss=0.06770285993814468
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2146 | LCC = 0.9696 | SRCC = 0.9693 ]
|
| 38 |
+
INFO:root:Average loss=0.06310671600699425
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1457 | LCC = 0.9690 | SRCC = 0.9686 ]
|
| 41 |
+
INFO:root:Average loss=0.056994130730628964
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1926 | LCC = 0.9696 | SRCC = 0.9701 ]
|
| 44 |
+
INFO:root:Average loss=0.054625910580158234
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1578 | LCC = 0.9700 | SRCC = 0.9691 ]
|
| 47 |
+
INFO:root:Average loss=0.051796904265880586
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2039 | LCC = 0.9675 | SRCC = 0.9674 ]
|
| 50 |
+
INFO:root:Average loss=0.04977180987596512
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1849 | LCC = 0.9679 | SRCC = 0.9680 ]
|
| 53 |
+
INFO:root:Average loss=0.044429125517606735
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1734 | LCC = 0.9703 | SRCC = 0.9696 ]
|
| 56 |
+
INFO:root:Average loss=0.043082465261220934
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1581 | LCC = 0.9691 | SRCC = 0.9685 ]
|
| 59 |
+
INFO:root:Average loss=0.0434044725894928
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1267 | LCC = 0.9706 | SRCC = 0.9701 ]
|
| 62 |
+
INFO:root:Average loss=0.03978926730155945
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1833 | LCC = 0.9699 | SRCC = 0.9694 ]
|
| 65 |
+
INFO:root:Average loss=0.03919456271827221
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1736 | LCC = 0.9704 | SRCC = 0.9701 ]
|
| 68 |
+
INFO:root:Average loss=0.0359546305090189
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1468 | LCC = 0.9701 | SRCC = 0.9694 ]
|
| 71 |
+
INFO:root:Average loss=0.03708182345330715
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1795 | LCC = 0.9711 | SRCC = 0.9705 ]
|
| 74 |
+
INFO:root:Average loss=0.035054993987083435
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1552 | LCC = 0.9707 | SRCC = 0.9700 ]
|
| 77 |
+
INFO:root:Average loss=0.03389630497992039
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1574 | LCC = 0.9701 | SRCC = 0.9700 ]
|
| 80 |
+
INFO:root:Average loss=0.03245997132360935
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1844 | LCC = 0.9701 | SRCC = 0.9694 ]
|
| 83 |
+
INFO:root:Average loss=0.030413863480091095
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1572 | LCC = 0.9710 | SRCC = 0.9705 ]
|
| 86 |
+
INFO:root:Average loss=0.031395073607563974
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1631 | LCC = 0.9700 | SRCC = 0.9689 ]
|
| 89 |
+
INFO:root:Average loss=0.02919469544291496
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1755 | LCC = 0.9701 | SRCC = 0.9692 ]
|
| 92 |
+
INFO:root:Average loss=0.02769296584278345
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1592 | LCC = 0.9714 | SRCC = 0.9706 ]
|
| 95 |
+
INFO:root:Average loss=0.026416000045835972
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1723 | LCC = 0.9710 | SRCC = 0.9704 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1614 | LCC = 0.9701 | SRCC = 0.9701 ]
|
results_tencent/w2v2_xlsr_300m/layer4/run1/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer4/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:918f90d9cc4ede4fafb0658cd8360bca794ee85c2d36ca61193bdad4496030fd
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer4/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bf0f34a9b9d10698f48e5cbde6a7f30fdeccc79700ecd089fcd2ec5862f0f330
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer4/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.3539437585473061
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2316 | LCC = 0.9463 | SRCC = 0.9447 ]
|
| 8 |
+
INFO:root:Average loss=0.1384958703517914
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2892 | LCC = 0.9547 | SRCC = 0.9543 ]
|
| 11 |
+
INFO:root:Average loss=0.12476984417438507
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2154 | LCC = 0.9593 | SRCC = 0.9591 ]
|
| 14 |
+
INFO:root:Average loss=0.10756545868515968
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1943 | LCC = 0.9620 | SRCC = 0.9620 ]
|
| 17 |
+
INFO:root:Average loss=0.09650271216034889
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2152 | LCC = 0.9637 | SRCC = 0.9640 ]
|
| 20 |
+
INFO:root:Average loss=0.08482027754187584
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1902 | LCC = 0.9665 | SRCC = 0.9656 ]
|
| 23 |
+
INFO:root:Average loss=0.0782379894554615
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2176 | LCC = 0.9645 | SRCC = 0.9638 ]
|
| 26 |
+
INFO:root:Average loss=0.07717566844820976
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1921 | LCC = 0.9676 | SRCC = 0.9668 ]
|
| 29 |
+
INFO:root:Average loss=0.07340138745307923
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1647 | LCC = 0.9670 | SRCC = 0.9672 ]
|
| 32 |
+
INFO:root:Average loss=0.06561676475405694
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2143 | LCC = 0.9690 | SRCC = 0.9686 ]
|
| 35 |
+
INFO:root:Average loss=0.06151456007361412
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2067 | LCC = 0.9689 | SRCC = 0.9683 ]
|
| 38 |
+
INFO:root:Average loss=0.06133492562174797
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1873 | LCC = 0.9685 | SRCC = 0.9681 ]
|
| 41 |
+
INFO:root:Average loss=0.057051753878593445
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1582 | LCC = 0.9691 | SRCC = 0.9681 ]
|
| 44 |
+
INFO:root:Average loss=0.055188780531287195
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2032 | LCC = 0.9690 | SRCC = 0.9684 ]
|
| 47 |
+
INFO:root:Average loss=0.05235090309381485
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1706 | LCC = 0.9701 | SRCC = 0.9689 ]
|
| 50 |
+
INFO:root:Average loss=0.04848231323063373
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1827 | LCC = 0.9699 | SRCC = 0.9689 ]
|
| 53 |
+
INFO:root:Average loss=0.048031270012259486
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1583 | LCC = 0.9691 | SRCC = 0.9689 ]
|
| 56 |
+
INFO:root:Average loss=0.045001750349998475
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1901 | LCC = 0.9692 | SRCC = 0.9680 ]
|
| 59 |
+
INFO:root:Average loss=0.04325820714235306
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1839 | LCC = 0.9689 | SRCC = 0.9685 ]
|
| 62 |
+
INFO:root:Average loss=0.04364639917016029
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1854 | LCC = 0.9699 | SRCC = 0.9693 ]
|
| 65 |
+
INFO:root:Average loss=0.037001041427254674
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2028 | LCC = 0.9692 | SRCC = 0.9684 ]
|
| 68 |
+
INFO:root:Average loss=0.03794203832745552
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1609 | LCC = 0.9695 | SRCC = 0.9689 ]
|
| 71 |
+
INFO:root:Average loss=0.035563049912452696
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1807 | LCC = 0.9696 | SRCC = 0.9685 ]
|
| 74 |
+
INFO:root:Average loss=0.034318716898560524
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1774 | LCC = 0.9681 | SRCC = 0.9675 ]
|
| 77 |
+
INFO:root:Average loss=0.033422301664948466
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1669 | LCC = 0.9702 | SRCC = 0.9695 ]
|
| 80 |
+
INFO:root:Average loss=0.033809678971767426
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1618 | LCC = 0.9696 | SRCC = 0.9687 ]
|
| 83 |
+
INFO:root:Average loss=0.03175129644572735
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1680 | LCC = 0.9699 | SRCC = 0.9695 ]
|
| 86 |
+
INFO:root:Average loss=0.032571035385131834
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1877 | LCC = 0.9698 | SRCC = 0.9689 ]
|
| 89 |
+
INFO:root:Average loss=0.03213731575012207
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1871 | LCC = 0.9694 | SRCC = 0.9686 ]
|
| 92 |
+
INFO:root:Average loss=0.028882778719067574
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1790 | LCC = 0.9699 | SRCC = 0.9688 ]
|
| 95 |
+
INFO:root:Average loss=0.02779819478094578
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2094 | LCC = 0.9689 | SRCC = 0.9680 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1690 | LCC = 0.9696 | SRCC = 0.9695 ]
|
results_tencent/w2v2_xlsr_300m/layer4/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer4/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:98f495e1ff221ff61745f039741a8891784fa8ee6b1b76ce4231bd693429eb0f
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer4/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:806938213893bb7a5d2fc6b8204779591bc184b270209e9de8bae042cd2df46e
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer4/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.32357763147354124
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2447 | LCC = 0.9479 | SRCC = 0.9466 ]
|
| 8 |
+
INFO:root:Average loss=0.13941112178564072
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.1826 | LCC = 0.9575 | SRCC = 0.9580 ]
|
| 11 |
+
INFO:root:Average loss=0.11757403671741486
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1995 | LCC = 0.9617 | SRCC = 0.9618 ]
|
| 14 |
+
INFO:root:Average loss=0.10048690873384476
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1944 | LCC = 0.9629 | SRCC = 0.9618 ]
|
| 17 |
+
INFO:root:Average loss=0.09427812054753304
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1960 | LCC = 0.9649 | SRCC = 0.9644 ]
|
| 20 |
+
INFO:root:Average loss=0.08512272840738297
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1736 | LCC = 0.9666 | SRCC = 0.9663 ]
|
| 23 |
+
INFO:root:Average loss=0.07750110748410224
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1773 | LCC = 0.9687 | SRCC = 0.9681 ]
|
| 26 |
+
INFO:root:Average loss=0.0740789364874363
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1501 | LCC = 0.9660 | SRCC = 0.9673 ]
|
| 29 |
+
INFO:root:Average loss=0.07480200892686843
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1988 | LCC = 0.9689 | SRCC = 0.9682 ]
|
| 32 |
+
INFO:root:Average loss=0.06501302102208137
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1840 | LCC = 0.9699 | SRCC = 0.9694 ]
|
| 35 |
+
INFO:root:Average loss=0.06238117054104805
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1899 | LCC = 0.9701 | SRCC = 0.9689 ]
|
| 38 |
+
INFO:root:Average loss=0.05812867943942547
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1971 | LCC = 0.9698 | SRCC = 0.9689 ]
|
| 41 |
+
INFO:root:Average loss=0.05790045550465584
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1662 | LCC = 0.9693 | SRCC = 0.9683 ]
|
| 44 |
+
INFO:root:Average loss=0.05575731548666954
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1630 | LCC = 0.9706 | SRCC = 0.9698 ]
|
| 47 |
+
INFO:root:Average loss=0.05430857688188553
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1771 | LCC = 0.9698 | SRCC = 0.9695 ]
|
| 50 |
+
INFO:root:Average loss=0.04787358658015728
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2146 | LCC = 0.9681 | SRCC = 0.9680 ]
|
| 53 |
+
INFO:root:Average loss=0.048314673706889154
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1617 | LCC = 0.9711 | SRCC = 0.9702 ]
|
| 56 |
+
INFO:root:Average loss=0.04674104622006416
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1683 | LCC = 0.9716 | SRCC = 0.9705 ]
|
| 59 |
+
INFO:root:Average loss=0.04354212300479412
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1583 | LCC = 0.9708 | SRCC = 0.9702 ]
|
| 62 |
+
INFO:root:Average loss=0.03950455245375633
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1576 | LCC = 0.9702 | SRCC = 0.9697 ]
|
| 65 |
+
INFO:root:Average loss=0.03978278426826
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1449 | LCC = 0.9706 | SRCC = 0.9701 ]
|
| 68 |
+
INFO:root:Average loss=0.03667403574287891
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1688 | LCC = 0.9707 | SRCC = 0.9698 ]
|
| 71 |
+
INFO:root:Average loss=0.03677007168531418
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1630 | LCC = 0.9719 | SRCC = 0.9709 ]
|
| 74 |
+
INFO:root:Average loss=0.037206030666828156
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1850 | LCC = 0.9712 | SRCC = 0.9704 ]
|
| 77 |
+
INFO:root:Average loss=0.035217587411403656
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1671 | LCC = 0.9717 | SRCC = 0.9705 ]
|
| 80 |
+
INFO:root:Average loss=0.03294680121541023
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1450 | LCC = 0.9711 | SRCC = 0.9702 ]
|
| 83 |
+
INFO:root:Average loss=0.031606700390577315
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1594 | LCC = 0.9714 | SRCC = 0.9705 ]
|
| 86 |
+
INFO:root:Average loss=0.030644863337278366
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1525 | LCC = 0.9701 | SRCC = 0.9688 ]
|
| 89 |
+
INFO:root:Average loss=0.030930347740650177
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1553 | LCC = 0.9716 | SRCC = 0.9707 ]
|
| 92 |
+
INFO:root:Average loss=0.030007797315716744
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1727 | LCC = 0.9712 | SRCC = 0.9702 ]
|
| 95 |
+
INFO:root:Average loss=0.02821310679614544
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1762 | LCC = 0.9710 | SRCC = 0.9700 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1662 | LCC = 0.9700 | SRCC = 0.9697 ]
|
results_tencent/w2v2_xlsr_300m/layer4/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on Tencent 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 |
+
Tencent:
|
| 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 = @Tencent
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_tencent/w2v2_xlsr_300m/layer4/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f990918d7cdf395d8d730c19d765a27e43c3a2fc4181de8f68d4313921c721a2
|
| 3 |
+
size 3403339
|
results_tencent/w2v2_xlsr_300m/layer4/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a54c9aa6d917e1900690447e108bdcd499123c77003342eab4d9045a9ec4d9fa
|
| 3 |
+
size 3405062
|
results_tencent/w2v2_xlsr_300m/layer4/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 8000
|
| 2 |
+
INFO:root:Num val speech clips: 2000
|
| 3 |
+
INFO:root:Num test speech clips: 1563
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.3493855581283569
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2403 | LCC = 0.9463 | SRCC = 0.9453 ]
|
| 8 |
+
INFO:root:Average loss=0.15164289849996568
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.1953 | LCC = 0.9565 | SRCC = 0.9561 ]
|
| 11 |
+
INFO:root:Average loss=0.11794343787431717
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2122 | LCC = 0.9624 | SRCC = 0.9619 ]
|
| 14 |
+
INFO:root:Average loss=0.10819954192638397
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2554 | LCC = 0.9632 | SRCC = 0.9631 ]
|
| 17 |
+
INFO:root:Average loss=0.09795257690548896
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1999 | LCC = 0.9662 | SRCC = 0.9657 ]
|
| 20 |
+
INFO:root:Average loss=0.08717562344670296
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1745 | LCC = 0.9669 | SRCC = 0.9668 ]
|
| 23 |
+
INFO:root:Average loss=0.07944501149654389
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1817 | LCC = 0.9689 | SRCC = 0.9681 ]
|
| 26 |
+
INFO:root:Average loss=0.07397128212451935
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2258 | LCC = 0.9671 | SRCC = 0.9679 ]
|
| 29 |
+
INFO:root:Average loss=0.0712431848347187
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1866 | LCC = 0.9688 | SRCC = 0.9682 ]
|
| 32 |
+
INFO:root:Average loss=0.07080922010540962
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2336 | LCC = 0.9699 | SRCC = 0.9692 ]
|
| 35 |
+
INFO:root:Average loss=0.06404505583643913
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1647 | LCC = 0.9697 | SRCC = 0.9687 ]
|
| 38 |
+
INFO:root:Average loss=0.05922862908244133
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1842 | LCC = 0.9693 | SRCC = 0.9689 ]
|
| 41 |
+
INFO:root:Average loss=0.06013742625713348
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1932 | LCC = 0.9693 | SRCC = 0.9698 ]
|
| 44 |
+
INFO:root:Average loss=0.05613620123267174
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1834 | LCC = 0.9711 | SRCC = 0.9700 ]
|
| 47 |
+
INFO:root:Average loss=0.05181649950146675
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1994 | LCC = 0.9698 | SRCC = 0.9690 ]
|
| 50 |
+
INFO:root:Average loss=0.05046484811604023
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1915 | LCC = 0.9699 | SRCC = 0.9692 ]
|
| 53 |
+
INFO:root:Average loss=0.048300379991531375
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1946 | LCC = 0.9709 | SRCC = 0.9701 ]
|
| 56 |
+
INFO:root:Average loss=0.04513438878953457
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1508 | LCC = 0.9704 | SRCC = 0.9697 ]
|
| 59 |
+
INFO:root:Average loss=0.04444257880747318
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2046 | LCC = 0.9697 | SRCC = 0.9688 ]
|
| 62 |
+
INFO:root:Average loss=0.04042323687672615
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1672 | LCC = 0.9708 | SRCC = 0.9698 ]
|
| 65 |
+
INFO:root:Average loss=0.03882869379222393
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1756 | LCC = 0.9701 | SRCC = 0.9692 ]
|
| 68 |
+
INFO:root:Average loss=0.037710220754146574
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1897 | LCC = 0.9702 | SRCC = 0.9692 ]
|
| 71 |
+
INFO:root:Average loss=0.03739005798101425
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1870 | LCC = 0.9713 | SRCC = 0.9704 ]
|
| 74 |
+
INFO:root:Average loss=0.03796876099705696
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1810 | LCC = 0.9695 | SRCC = 0.9682 ]
|
| 77 |
+
INFO:root:Average loss=0.033981205120682714
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1909 | LCC = 0.9694 | SRCC = 0.9685 ]
|
| 80 |
+
INFO:root:Average loss=0.03287234637141228
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1695 | LCC = 0.9689 | SRCC = 0.9682 ]
|
| 83 |
+
INFO:root:Average loss=0.031535907059907914
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1660 | LCC = 0.9708 | SRCC = 0.9703 ]
|
| 86 |
+
INFO:root:Average loss=0.03071501253545284
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1736 | LCC = 0.9712 | SRCC = 0.9701 ]
|
| 89 |
+
INFO:root:Average loss=0.027936208471655847
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1413 | LCC = 0.9698 | SRCC = 0.9689 ]
|
| 92 |
+
INFO:root:Average loss=0.028565396949648856
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1831 | LCC = 0.9693 | SRCC = 0.9683 ]
|
| 95 |
+
INFO:root:Average loss=0.03051776148378849
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1636 | LCC = 0.9695 | SRCC = 0.9689 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1884 | LCC = 0.9699 | SRCC = 0.9695 ]
|