Upload SSL Layer MOS checkpoints (part 19)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_tencent/w2v2_xlsr_1b/layer34/run1/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run1/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer34/run2/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer34/run2/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run2/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run2/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer34/run3/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer34/run3/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run3/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run3/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer34/run4/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer34/run4/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run4/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer34/run4/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer36/run0/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer36/run0/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run0/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run0/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer36/run1/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer36/run1/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run1/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run1/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer36/run2/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer36/run2/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run2/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run2/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer36/run3/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer36/run3/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run3/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run3/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer36/run4/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer36/run4/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run4/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer36/run4/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer38/run0/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer38/run0/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run0/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run0/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer38/run1/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer38/run1/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run1/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run1/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer38/run2/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer38/run2/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run2/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run2/train.log +99 -0
- results_tencent/w2v2_xlsr_1b/layer38/run3/config.gin +31 -0
- results_tencent/w2v2_xlsr_1b/layer38/run3/model.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run3/model_best.pt +3 -0
- results_tencent/w2v2_xlsr_1b/layer38/run3/train.log +99 -0
results_tencent/w2v2_xlsr_1b/layer34/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:daafc5d61a19dfb697bb03c6ff44d9b58fcf682917efaae08ff0d19d446c8025
|
| 3 |
+
size 4060422
|
results_tencent/w2v2_xlsr_1b/layer34/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.4902742062211037
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2398 | LCC = 0.9484 | SRCC = 0.9459 ]
|
| 8 |
+
INFO:root:Average loss=0.14633447909355163
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2604 | LCC = 0.9567 | SRCC = 0.9579 ]
|
| 11 |
+
INFO:root:Average loss=0.12721437513828276
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1669 | LCC = 0.9626 | SRCC = 0.9621 ]
|
| 14 |
+
INFO:root:Average loss=0.1046346583366394
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1866 | LCC = 0.9657 | SRCC = 0.9651 ]
|
| 17 |
+
INFO:root:Average loss=0.09733275064826012
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1638 | LCC = 0.9664 | SRCC = 0.9660 ]
|
| 20 |
+
INFO:root:Average loss=0.08789659485220909
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1676 | LCC = 0.9684 | SRCC = 0.9675 ]
|
| 23 |
+
INFO:root:Average loss=0.0830150817334652
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1765 | LCC = 0.9677 | SRCC = 0.9679 ]
|
| 26 |
+
INFO:root:Average loss=0.07978170904517173
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1479 | LCC = 0.9699 | SRCC = 0.9690 ]
|
| 29 |
+
INFO:root:Average loss=0.07524063646793365
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1509 | LCC = 0.9680 | SRCC = 0.9665 ]
|
| 32 |
+
INFO:root:Average loss=0.07233297297358512
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1749 | LCC = 0.9703 | SRCC = 0.9698 ]
|
| 35 |
+
INFO:root:Average loss=0.06812826901674271
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1709 | LCC = 0.9708 | SRCC = 0.9702 ]
|
| 38 |
+
INFO:root:Average loss=0.06281923732161522
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1504 | LCC = 0.9711 | SRCC = 0.9704 ]
|
| 41 |
+
INFO:root:Average loss=0.06082053729891777
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1516 | LCC = 0.9705 | SRCC = 0.9699 ]
|
| 44 |
+
INFO:root:Average loss=0.05907168531417847
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1739 | LCC = 0.9711 | SRCC = 0.9702 ]
|
| 47 |
+
INFO:root:Average loss=0.05746838785707951
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1801 | LCC = 0.9712 | SRCC = 0.9697 ]
|
| 50 |
+
INFO:root:Average loss=0.05068235886096954
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1560 | LCC = 0.9709 | SRCC = 0.9695 ]
|
| 53 |
+
INFO:root:Average loss=0.05124501751363277
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1653 | LCC = 0.9714 | SRCC = 0.9705 ]
|
| 56 |
+
INFO:root:Average loss=0.04573335160315037
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1542 | LCC = 0.9713 | SRCC = 0.9702 ]
|
| 59 |
+
INFO:root:Average loss=0.045032204672694205
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1484 | LCC = 0.9710 | SRCC = 0.9701 ]
|
| 62 |
+
INFO:root:Average loss=0.04412492543458939
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1536 | LCC = 0.9713 | SRCC = 0.9705 ]
|
| 65 |
+
INFO:root:Average loss=0.04120438221096993
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1473 | LCC = 0.9714 | SRCC = 0.9704 ]
|
| 68 |
+
INFO:root:Average loss=0.04032501465082169
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1548 | LCC = 0.9712 | SRCC = 0.9700 ]
|
| 71 |
+
INFO:root:Average loss=0.03921673908829689
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1411 | LCC = 0.9723 | SRCC = 0.9711 ]
|
| 74 |
+
INFO:root:Average loss=0.03591878201067448
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1573 | LCC = 0.9710 | SRCC = 0.9699 ]
|
| 77 |
+
INFO:root:Average loss=0.03771319682896137
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1470 | LCC = 0.9718 | SRCC = 0.9706 ]
|
| 80 |
+
INFO:root:Average loss=0.03572441041469574
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1560 | LCC = 0.9713 | SRCC = 0.9704 ]
|
| 83 |
+
INFO:root:Average loss=0.03770526441931724
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1440 | LCC = 0.9716 | SRCC = 0.9704 ]
|
| 86 |
+
INFO:root:Average loss=0.03133926630020142
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1340 | LCC = 0.9704 | SRCC = 0.9698 ]
|
| 89 |
+
INFO:root:Average loss=0.031310425609350205
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1377 | LCC = 0.9717 | SRCC = 0.9707 ]
|
| 92 |
+
INFO:root:Average loss=0.030254511252045632
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1485 | LCC = 0.9717 | SRCC = 0.9703 ]
|
| 95 |
+
INFO:root:Average loss=0.028796360448002815
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1493 | LCC = 0.9713 | SRCC = 0.9695 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1423 | LCC = 0.9713 | SRCC = 0.9707 ]
|
results_tencent/w2v2_xlsr_1b/layer34/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_1b/layer34/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:abfe23df20637d2229ab176812129e6bd0599c366e2b20c580fa535c2a9ebdea
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer34/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:53f2f520bb57a11f1f04be9999c6d7d7444c4ec5e08d7a29296dec1b317ab0ff
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer34/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.5116374804973602
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2605 | LCC = 0.9452 | SRCC = 0.9420 ]
|
| 8 |
+
INFO:root:Average loss=0.17314399302005767
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2513 | LCC = 0.9569 | SRCC = 0.9561 ]
|
| 11 |
+
INFO:root:Average loss=0.12890104061365126
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2070 | LCC = 0.9627 | SRCC = 0.9625 ]
|
| 14 |
+
INFO:root:Average loss=0.11005250978469848
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2378 | LCC = 0.9647 | SRCC = 0.9645 ]
|
| 17 |
+
INFO:root:Average loss=0.0985694120824337
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1762 | LCC = 0.9658 | SRCC = 0.9656 ]
|
| 20 |
+
INFO:root:Average loss=0.09373472476005554
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2205 | LCC = 0.9663 | SRCC = 0.9663 ]
|
| 23 |
+
INFO:root:Average loss=0.08342433172464371
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1952 | LCC = 0.9666 | SRCC = 0.9667 ]
|
| 26 |
+
INFO:root:Average loss=0.08070914933085442
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1980 | LCC = 0.9677 | SRCC = 0.9670 ]
|
| 29 |
+
INFO:root:Average loss=0.0756270144879818
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1873 | LCC = 0.9693 | SRCC = 0.9685 ]
|
| 32 |
+
INFO:root:Average loss=0.07366727724671364
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1528 | LCC = 0.9683 | SRCC = 0.9689 ]
|
| 35 |
+
INFO:root:Average loss=0.06635770091414452
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1696 | LCC = 0.9685 | SRCC = 0.9679 ]
|
| 38 |
+
INFO:root:Average loss=0.06216112345457077
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1881 | LCC = 0.9706 | SRCC = 0.9695 ]
|
| 41 |
+
INFO:root:Average loss=0.06074040991067886
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1688 | LCC = 0.9704 | SRCC = 0.9701 ]
|
| 44 |
+
INFO:root:Average loss=0.060610357597470285
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1765 | LCC = 0.9706 | SRCC = 0.9696 ]
|
| 47 |
+
INFO:root:Average loss=0.055302672609686854
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1679 | LCC = 0.9703 | SRCC = 0.9698 ]
|
| 50 |
+
INFO:root:Average loss=0.05374186682701111
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1651 | LCC = 0.9705 | SRCC = 0.9694 ]
|
| 53 |
+
INFO:root:Average loss=0.05052183696627617
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1975 | LCC = 0.9703 | SRCC = 0.9693 ]
|
| 56 |
+
INFO:root:Average loss=0.05052878153324127
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1635 | LCC = 0.9702 | SRCC = 0.9697 ]
|
| 59 |
+
INFO:root:Average loss=0.048052635431289674
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1695 | LCC = 0.9715 | SRCC = 0.9701 ]
|
| 62 |
+
INFO:root:Average loss=0.04415951931476593
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1717 | LCC = 0.9712 | SRCC = 0.9698 ]
|
| 65 |
+
INFO:root:Average loss=0.04442810708284378
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1783 | LCC = 0.9718 | SRCC = 0.9707 ]
|
| 68 |
+
INFO:root:Average loss=0.039961826995015146
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1973 | LCC = 0.9715 | SRCC = 0.9710 ]
|
| 71 |
+
INFO:root:Average loss=0.03955658057332039
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1821 | LCC = 0.9697 | SRCC = 0.9684 ]
|
| 74 |
+
INFO:root:Average loss=0.038611474484205245
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1539 | LCC = 0.9718 | SRCC = 0.9708 ]
|
| 77 |
+
INFO:root:Average loss=0.0374235537648201
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1552 | LCC = 0.9692 | SRCC = 0.9690 ]
|
| 80 |
+
INFO:root:Average loss=0.035631714433431626
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1569 | LCC = 0.9705 | SRCC = 0.9703 ]
|
| 83 |
+
INFO:root:Average loss=0.034970214545726776
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1726 | LCC = 0.9710 | SRCC = 0.9700 ]
|
| 86 |
+
INFO:root:Average loss=0.03262574368715286
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1751 | LCC = 0.9700 | SRCC = 0.9691 ]
|
| 89 |
+
INFO:root:Average loss=0.03202556210756302
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1588 | LCC = 0.9708 | SRCC = 0.9702 ]
|
| 92 |
+
INFO:root:Average loss=0.030695222318172455
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1656 | LCC = 0.9713 | SRCC = 0.9699 ]
|
| 95 |
+
INFO:root:Average loss=0.03096596737205982
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1641 | LCC = 0.9709 | SRCC = 0.9698 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1746 | LCC = 0.9733 | SRCC = 0.9725 ]
|
results_tencent/w2v2_xlsr_1b/layer34/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_1b/layer34/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:420f117e0bef9ec02074948b2e72ab0273af756c09b6da9cae22f932e33cbbea
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer34/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:25c7f5658c0535e04cad24513dd3de1ef2bf2dcadeea569d6b8ac5bf8c5a313f
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer34/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.42322697162628176
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2565 | LCC = 0.9407 | SRCC = 0.9397 ]
|
| 8 |
+
INFO:root:Average loss=0.1641413117647171
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2434 | LCC = 0.9571 | SRCC = 0.9567 ]
|
| 11 |
+
INFO:root:Average loss=0.12665840870141984
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2400 | LCC = 0.9621 | SRCC = 0.9604 ]
|
| 14 |
+
INFO:root:Average loss=0.11475816321372986
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2087 | LCC = 0.9644 | SRCC = 0.9643 ]
|
| 17 |
+
INFO:root:Average loss=0.10551588690280914
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1569 | LCC = 0.9667 | SRCC = 0.9660 ]
|
| 20 |
+
INFO:root:Average loss=0.09330324819684029
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1858 | LCC = 0.9684 | SRCC = 0.9679 ]
|
| 23 |
+
INFO:root:Average loss=0.08807834824919701
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1634 | LCC = 0.9681 | SRCC = 0.9683 ]
|
| 26 |
+
INFO:root:Average loss=0.07908114370703698
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1693 | LCC = 0.9696 | SRCC = 0.9692 ]
|
| 29 |
+
INFO:root:Average loss=0.07634212982654572
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1892 | LCC = 0.9700 | SRCC = 0.9697 ]
|
| 32 |
+
INFO:root:Average loss=0.07429390037059784
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2025 | LCC = 0.9702 | SRCC = 0.9699 ]
|
| 35 |
+
INFO:root:Average loss=0.0714883383512497
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1644 | LCC = 0.9706 | SRCC = 0.9700 ]
|
| 38 |
+
INFO:root:Average loss=0.06761336946487427
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1512 | LCC = 0.9696 | SRCC = 0.9701 ]
|
| 41 |
+
INFO:root:Average loss=0.06911549091339111
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1901 | LCC = 0.9710 | SRCC = 0.9702 ]
|
| 44 |
+
INFO:root:Average loss=0.05884688803553581
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1729 | LCC = 0.9717 | SRCC = 0.9707 ]
|
| 47 |
+
INFO:root:Average loss=0.05701669503748417
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1604 | LCC = 0.9712 | SRCC = 0.9702 ]
|
| 50 |
+
INFO:root:Average loss=0.05438076056540012
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1503 | LCC = 0.9723 | SRCC = 0.9714 ]
|
| 53 |
+
INFO:root:Average loss=0.052790969014167785
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1933 | LCC = 0.9715 | SRCC = 0.9717 ]
|
| 56 |
+
INFO:root:Average loss=0.05152842557430267
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1698 | LCC = 0.9725 | SRCC = 0.9717 ]
|
| 59 |
+
INFO:root:Average loss=0.048267154276371
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1493 | LCC = 0.9708 | SRCC = 0.9697 ]
|
| 62 |
+
INFO:root:Average loss=0.04684811107814312
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1916 | LCC = 0.9712 | SRCC = 0.9707 ]
|
| 65 |
+
INFO:root:Average loss=0.044876496985554695
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1547 | LCC = 0.9724 | SRCC = 0.9712 ]
|
| 68 |
+
INFO:root:Average loss=0.04433149087429047
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1653 | LCC = 0.9709 | SRCC = 0.9699 ]
|
| 71 |
+
INFO:root:Average loss=0.04125556680560112
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1612 | LCC = 0.9724 | SRCC = 0.9713 ]
|
| 74 |
+
INFO:root:Average loss=0.03947563420236111
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1622 | LCC = 0.9726 | SRCC = 0.9717 ]
|
| 77 |
+
INFO:root:Average loss=0.036119799092411996
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1686 | LCC = 0.9722 | SRCC = 0.9709 ]
|
| 80 |
+
INFO:root:Average loss=0.036942376121878624
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1474 | LCC = 0.9721 | SRCC = 0.9712 ]
|
| 83 |
+
INFO:root:Average loss=0.03647490967810154
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1468 | LCC = 0.9715 | SRCC = 0.9712 ]
|
| 86 |
+
INFO:root:Average loss=0.03421395343542099
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1701 | LCC = 0.9723 | SRCC = 0.9713 ]
|
| 89 |
+
INFO:root:Average loss=0.032851971536874774
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1552 | LCC = 0.9720 | SRCC = 0.9710 ]
|
| 92 |
+
INFO:root:Average loss=0.03334848453104496
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1555 | LCC = 0.9725 | SRCC = 0.9713 ]
|
| 95 |
+
INFO:root:Average loss=0.0340045747384429
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1596 | LCC = 0.9725 | SRCC = 0.9715 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1614 | LCC = 0.9727 | SRCC = 0.9722 ]
|
results_tencent/w2v2_xlsr_1b/layer34/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_1b/layer34/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:df6558a0a0b1b3e9512c07a3f4084d8d6897db66a5edc40d4d541dd8dd46095b
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer34/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c98a20d728561cb7c5b8562fef7f4f43af97c06067c01bf69c4e5d3837cc3a8e
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer34/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.36400548738241195
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2443 | LCC = 0.9433 | SRCC = 0.9420 ]
|
| 8 |
+
INFO:root:Average loss=0.15119187492132186
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2264 | LCC = 0.9552 | SRCC = 0.9540 ]
|
| 11 |
+
INFO:root:Average loss=0.12893920147418975
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2173 | LCC = 0.9599 | SRCC = 0.9593 ]
|
| 14 |
+
INFO:root:Average loss=0.11330500611662865
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2049 | LCC = 0.9610 | SRCC = 0.9615 ]
|
| 17 |
+
INFO:root:Average loss=0.09980326175689697
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1914 | LCC = 0.9647 | SRCC = 0.9642 ]
|
| 20 |
+
INFO:root:Average loss=0.08943218106031418
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1918 | LCC = 0.9655 | SRCC = 0.9650 ]
|
| 23 |
+
INFO:root:Average loss=0.0829474957883358
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1767 | LCC = 0.9657 | SRCC = 0.9655 ]
|
| 26 |
+
INFO:root:Average loss=0.08013744083046913
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1753 | LCC = 0.9635 | SRCC = 0.9643 ]
|
| 29 |
+
INFO:root:Average loss=0.07496358072757721
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1987 | LCC = 0.9680 | SRCC = 0.9674 ]
|
| 32 |
+
INFO:root:Average loss=0.0705914244055748
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1751 | LCC = 0.9681 | SRCC = 0.9674 ]
|
| 35 |
+
INFO:root:Average loss=0.06566153863072395
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1477 | LCC = 0.9667 | SRCC = 0.9675 ]
|
| 38 |
+
INFO:root:Average loss=0.06612320005893707
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1635 | LCC = 0.9683 | SRCC = 0.9677 ]
|
| 41 |
+
INFO:root:Average loss=0.06122040030360222
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1631 | LCC = 0.9681 | SRCC = 0.9678 ]
|
| 44 |
+
INFO:root:Average loss=0.05771169379353523
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1870 | LCC = 0.9687 | SRCC = 0.9677 ]
|
| 47 |
+
INFO:root:Average loss=0.05371959680318832
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1564 | LCC = 0.9684 | SRCC = 0.9674 ]
|
| 50 |
+
INFO:root:Average loss=0.05002111430466175
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2096 | LCC = 0.9673 | SRCC = 0.9676 ]
|
| 53 |
+
INFO:root:Average loss=0.050281098887324335
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2241 | LCC = 0.9676 | SRCC = 0.9676 ]
|
| 56 |
+
INFO:root:Average loss=0.048399248272180555
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1610 | LCC = 0.9685 | SRCC = 0.9677 ]
|
| 59 |
+
INFO:root:Average loss=0.04781105010211468
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1644 | LCC = 0.9691 | SRCC = 0.9683 ]
|
| 62 |
+
INFO:root:Average loss=0.045207404017448424
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1852 | LCC = 0.9687 | SRCC = 0.9672 ]
|
| 65 |
+
INFO:root:Average loss=0.04095040114223957
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1715 | LCC = 0.9685 | SRCC = 0.9679 ]
|
| 68 |
+
INFO:root:Average loss=0.03902539610862732
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1677 | LCC = 0.9688 | SRCC = 0.9678 ]
|
| 71 |
+
INFO:root:Average loss=0.038662344694137575
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1551 | LCC = 0.9690 | SRCC = 0.9681 ]
|
| 74 |
+
INFO:root:Average loss=0.03937659129500389
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1581 | LCC = 0.9686 | SRCC = 0.9677 ]
|
| 77 |
+
INFO:root:Average loss=0.0358126219958067
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1585 | LCC = 0.9679 | SRCC = 0.9673 ]
|
| 80 |
+
INFO:root:Average loss=0.03809351152181625
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1512 | LCC = 0.9686 | SRCC = 0.9682 ]
|
| 83 |
+
INFO:root:Average loss=0.034593225583434105
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1873 | LCC = 0.9682 | SRCC = 0.9672 ]
|
| 86 |
+
INFO:root:Average loss=0.03302263800799847
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1403 | LCC = 0.9678 | SRCC = 0.9671 ]
|
| 89 |
+
INFO:root:Average loss=0.03317894513905048
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1478 | LCC = 0.9679 | SRCC = 0.9678 ]
|
| 92 |
+
INFO:root:Average loss=0.031097458362579346
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1637 | LCC = 0.9684 | SRCC = 0.9677 ]
|
| 95 |
+
INFO:root:Average loss=0.029349580004811287
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1649 | LCC = 0.9693 | SRCC = 0.9685 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1617 | LCC = 0.9708 | SRCC = 0.9705 ]
|
results_tencent/w2v2_xlsr_1b/layer36/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_1b/layer36/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c3de9439ad96ddcb9aa14b1035b927b3b9a10843b32745befdea5e3c47d99832
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer36/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b33b6a2f48b87897d70f89deb210c93b16b419199afe32ae7815c2e3dab3f6e7
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer36/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.37084116476774215
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2625 | LCC = 0.9456 | SRCC = 0.9431 ]
|
| 8 |
+
INFO:root:Average loss=0.1466196327805519
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2394 | LCC = 0.9583 | SRCC = 0.9575 ]
|
| 11 |
+
INFO:root:Average loss=0.11966153651475907
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2065 | LCC = 0.9630 | SRCC = 0.9634 ]
|
| 14 |
+
INFO:root:Average loss=0.09997559505701065
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1902 | LCC = 0.9657 | SRCC = 0.9652 ]
|
| 17 |
+
INFO:root:Average loss=0.09148479464650154
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2435 | LCC = 0.9668 | SRCC = 0.9661 ]
|
| 20 |
+
INFO:root:Average loss=0.08862799698114396
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2219 | LCC = 0.9672 | SRCC = 0.9669 ]
|
| 23 |
+
INFO:root:Average loss=0.07906556087732315
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1955 | LCC = 0.9687 | SRCC = 0.9682 ]
|
| 26 |
+
INFO:root:Average loss=0.0771837994158268
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2188 | LCC = 0.9675 | SRCC = 0.9672 ]
|
| 29 |
+
INFO:root:Average loss=0.07137568792700767
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2150 | LCC = 0.9686 | SRCC = 0.9682 ]
|
| 32 |
+
INFO:root:Average loss=0.06630247557163238
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1683 | LCC = 0.9695 | SRCC = 0.9690 ]
|
| 35 |
+
INFO:root:Average loss=0.06341636164486408
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1664 | LCC = 0.9703 | SRCC = 0.9696 ]
|
| 38 |
+
INFO:root:Average loss=0.05839603883028031
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1710 | LCC = 0.9699 | SRCC = 0.9692 ]
|
| 41 |
+
INFO:root:Average loss=0.05651375888288021
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1611 | LCC = 0.9697 | SRCC = 0.9692 ]
|
| 44 |
+
INFO:root:Average loss=0.05258131051063537
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1918 | LCC = 0.9702 | SRCC = 0.9694 ]
|
| 47 |
+
INFO:root:Average loss=0.04937686942517758
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1657 | LCC = 0.9704 | SRCC = 0.9695 ]
|
| 50 |
+
INFO:root:Average loss=0.05143510685861111
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1500 | LCC = 0.9700 | SRCC = 0.9691 ]
|
| 53 |
+
INFO:root:Average loss=0.04847771035134792
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1584 | LCC = 0.9710 | SRCC = 0.9700 ]
|
| 56 |
+
INFO:root:Average loss=0.04459230253100395
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1822 | LCC = 0.9692 | SRCC = 0.9691 ]
|
| 59 |
+
INFO:root:Average loss=0.04214941976964474
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2095 | LCC = 0.9691 | SRCC = 0.9685 ]
|
| 62 |
+
INFO:root:Average loss=0.04159797143936157
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1819 | LCC = 0.9707 | SRCC = 0.9699 ]
|
| 65 |
+
INFO:root:Average loss=0.03915795786678791
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1957 | LCC = 0.9713 | SRCC = 0.9706 ]
|
| 68 |
+
INFO:root:Average loss=0.03757145771384239
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1768 | LCC = 0.9703 | SRCC = 0.9699 ]
|
| 71 |
+
INFO:root:Average loss=0.0359332632124424
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1579 | LCC = 0.9701 | SRCC = 0.9699 ]
|
| 74 |
+
INFO:root:Average loss=0.0351613128632307
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1718 | LCC = 0.9705 | SRCC = 0.9697 ]
|
| 77 |
+
INFO:root:Average loss=0.034543736264109615
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1587 | LCC = 0.9710 | SRCC = 0.9702 ]
|
| 80 |
+
INFO:root:Average loss=0.03201238319277763
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1876 | LCC = 0.9711 | SRCC = 0.9705 ]
|
| 83 |
+
INFO:root:Average loss=0.032399823412299154
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1858 | LCC = 0.9718 | SRCC = 0.9708 ]
|
| 86 |
+
INFO:root:Average loss=0.028996627867221833
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1793 | LCC = 0.9706 | SRCC = 0.9698 ]
|
| 89 |
+
INFO:root:Average loss=0.028090556278824806
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1949 | LCC = 0.9704 | SRCC = 0.9696 ]
|
| 92 |
+
INFO:root:Average loss=0.029231697738170623
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1894 | LCC = 0.9710 | SRCC = 0.9702 ]
|
| 95 |
+
INFO:root:Average loss=0.02756267496943474
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1757 | LCC = 0.9710 | SRCC = 0.9701 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1820 | LCC = 0.9722 | SRCC = 0.9716 ]
|
results_tencent/w2v2_xlsr_1b/layer36/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_1b/layer36/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aee5d8fe62198099bf1497d3c85640e789ab5b3bb0f3415b6090c86dd873b73b
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer36/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:336196f3a4b86b38c3dfcf82e27a83a10128a1321cb50b8379c4b9830c94ea53
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer36/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.39617545187473296
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2443 | LCC = 0.9484 | SRCC = 0.9477 ]
|
| 8 |
+
INFO:root:Average loss=0.14206642138957978
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2131 | LCC = 0.9578 | SRCC = 0.9575 ]
|
| 11 |
+
INFO:root:Average loss=0.11275172126293183
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2335 | LCC = 0.9636 | SRCC = 0.9629 ]
|
| 14 |
+
INFO:root:Average loss=0.09932926201820373
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1807 | LCC = 0.9650 | SRCC = 0.9646 ]
|
| 17 |
+
INFO:root:Average loss=0.09484005585312844
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2079 | LCC = 0.9673 | SRCC = 0.9666 ]
|
| 20 |
+
INFO:root:Average loss=0.08880963152647019
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2065 | LCC = 0.9680 | SRCC = 0.9672 ]
|
| 23 |
+
INFO:root:Average loss=0.08077442383766174
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1892 | LCC = 0.9688 | SRCC = 0.9683 ]
|
| 26 |
+
INFO:root:Average loss=0.07647999602556228
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1796 | LCC = 0.9693 | SRCC = 0.9685 ]
|
| 29 |
+
INFO:root:Average loss=0.06898756206035614
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1517 | LCC = 0.9705 | SRCC = 0.9699 ]
|
| 32 |
+
INFO:root:Average loss=0.0667625637948513
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1771 | LCC = 0.9703 | SRCC = 0.9697 ]
|
| 35 |
+
INFO:root:Average loss=0.06127203559875488
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1989 | LCC = 0.9700 | SRCC = 0.9698 ]
|
| 38 |
+
INFO:root:Average loss=0.06175964760780334
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1400 | LCC = 0.9700 | SRCC = 0.9695 ]
|
| 41 |
+
INFO:root:Average loss=0.058717340111732484
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1345 | LCC = 0.9696 | SRCC = 0.9703 ]
|
| 44 |
+
INFO:root:Average loss=0.05347913739085197
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1880 | LCC = 0.9705 | SRCC = 0.9694 ]
|
| 47 |
+
INFO:root:Average loss=0.051421783030033114
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1511 | LCC = 0.9712 | SRCC = 0.9705 ]
|
| 50 |
+
INFO:root:Average loss=0.04982315538823605
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1833 | LCC = 0.9716 | SRCC = 0.9706 ]
|
| 53 |
+
INFO:root:Average loss=0.04831311704218388
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1784 | LCC = 0.9709 | SRCC = 0.9706 ]
|
| 56 |
+
INFO:root:Average loss=0.046446365997195246
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1412 | LCC = 0.9709 | SRCC = 0.9709 ]
|
| 59 |
+
INFO:root:Average loss=0.04322810117900371
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1654 | LCC = 0.9710 | SRCC = 0.9698 ]
|
| 62 |
+
INFO:root:Average loss=0.0410259318202734
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1631 | LCC = 0.9712 | SRCC = 0.9704 ]
|
| 65 |
+
INFO:root:Average loss=0.03919383545219898
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1579 | LCC = 0.9706 | SRCC = 0.9696 ]
|
| 68 |
+
INFO:root:Average loss=0.03788477848470211
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1510 | LCC = 0.9714 | SRCC = 0.9703 ]
|
| 71 |
+
INFO:root:Average loss=0.03533661440014839
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1507 | LCC = 0.9709 | SRCC = 0.9700 ]
|
| 74 |
+
INFO:root:Average loss=0.03758320347964764
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1496 | LCC = 0.9712 | SRCC = 0.9703 ]
|
| 77 |
+
INFO:root:Average loss=0.0353399672806263
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1368 | LCC = 0.9708 | SRCC = 0.9701 ]
|
| 80 |
+
INFO:root:Average loss=0.035094783678650855
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1531 | LCC = 0.9719 | SRCC = 0.9710 ]
|
| 83 |
+
INFO:root:Average loss=0.031619231402873996
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1663 | LCC = 0.9724 | SRCC = 0.9713 ]
|
| 86 |
+
INFO:root:Average loss=0.029193493619561195
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1574 | LCC = 0.9715 | SRCC = 0.9707 ]
|
| 89 |
+
INFO:root:Average loss=0.03056273479759693
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1506 | LCC = 0.9712 | SRCC = 0.9702 ]
|
| 92 |
+
INFO:root:Average loss=0.028620364226400852
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1528 | LCC = 0.9706 | SRCC = 0.9701 ]
|
| 95 |
+
INFO:root:Average loss=0.029940906956791878
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1478 | LCC = 0.9709 | SRCC = 0.9701 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1640 | LCC = 0.9727 | SRCC = 0.9721 ]
|
results_tencent/w2v2_xlsr_1b/layer36/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_1b/layer36/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:674d0ab958539b5080b868effa331dd344e43dba7c26a2e04395e1e9e1966012
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer36/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aebe68c5e5e9efcc95728859252b408c01eaad77e6482d708669c331199afb96
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer36/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.39079192459583284
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2299 | LCC = 0.9470 | SRCC = 0.9461 ]
|
| 8 |
+
INFO:root:Average loss=0.14307945865392685
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.1950 | LCC = 0.9590 | SRCC = 0.9584 ]
|
| 11 |
+
INFO:root:Average loss=0.11326974377036095
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1983 | LCC = 0.9648 | SRCC = 0.9645 ]
|
| 14 |
+
INFO:root:Average loss=0.10428494682908059
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1877 | LCC = 0.9669 | SRCC = 0.9663 ]
|
| 17 |
+
INFO:root:Average loss=0.0958240289092064
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1907 | LCC = 0.9672 | SRCC = 0.9668 ]
|
| 20 |
+
INFO:root:Average loss=0.0870006841123104
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1801 | LCC = 0.9684 | SRCC = 0.9676 ]
|
| 23 |
+
INFO:root:Average loss=0.07758983084559441
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1796 | LCC = 0.9686 | SRCC = 0.9683 ]
|
| 26 |
+
INFO:root:Average loss=0.07353946301341056
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1427 | LCC = 0.9690 | SRCC = 0.9691 ]
|
| 29 |
+
INFO:root:Average loss=0.07332936948537827
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1850 | LCC = 0.9704 | SRCC = 0.9699 ]
|
| 32 |
+
INFO:root:Average loss=0.06465735644102097
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1705 | LCC = 0.9703 | SRCC = 0.9697 ]
|
| 35 |
+
INFO:root:Average loss=0.06295517334342003
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2091 | LCC = 0.9699 | SRCC = 0.9701 ]
|
| 38 |
+
INFO:root:Average loss=0.06022040620446205
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1607 | LCC = 0.9707 | SRCC = 0.9700 ]
|
| 41 |
+
INFO:root:Average loss=0.05581982740759849
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1366 | LCC = 0.9705 | SRCC = 0.9702 ]
|
| 44 |
+
INFO:root:Average loss=0.054982634544372556
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1969 | LCC = 0.9704 | SRCC = 0.9707 ]
|
| 47 |
+
INFO:root:Average loss=0.051424546033143996
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1768 | LCC = 0.9701 | SRCC = 0.9698 ]
|
| 50 |
+
INFO:root:Average loss=0.04961366702616215
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1573 | LCC = 0.9711 | SRCC = 0.9703 ]
|
| 53 |
+
INFO:root:Average loss=0.04846008837223053
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1439 | LCC = 0.9701 | SRCC = 0.9698 ]
|
| 56 |
+
INFO:root:Average loss=0.04522567541897297
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1720 | LCC = 0.9714 | SRCC = 0.9705 ]
|
| 59 |
+
INFO:root:Average loss=0.04323581247031689
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1728 | LCC = 0.9712 | SRCC = 0.9703 ]
|
| 62 |
+
INFO:root:Average loss=0.0419912174642086
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1694 | LCC = 0.9702 | SRCC = 0.9700 ]
|
| 65 |
+
INFO:root:Average loss=0.04146806475520134
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1639 | LCC = 0.9717 | SRCC = 0.9712 ]
|
| 68 |
+
INFO:root:Average loss=0.03771034741401672
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1709 | LCC = 0.9705 | SRCC = 0.9699 ]
|
| 71 |
+
INFO:root:Average loss=0.03704496978223324
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1590 | LCC = 0.9712 | SRCC = 0.9702 ]
|
| 74 |
+
INFO:root:Average loss=0.03763911803066731
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1647 | LCC = 0.9711 | SRCC = 0.9701 ]
|
| 77 |
+
INFO:root:Average loss=0.03491347037255764
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1566 | LCC = 0.9718 | SRCC = 0.9709 ]
|
| 80 |
+
INFO:root:Average loss=0.031737336680293084
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1625 | LCC = 0.9718 | SRCC = 0.9708 ]
|
| 83 |
+
INFO:root:Average loss=0.031511312022805216
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1631 | LCC = 0.9711 | SRCC = 0.9702 ]
|
| 86 |
+
INFO:root:Average loss=0.031939387530088424
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1582 | LCC = 0.9720 | SRCC = 0.9710 ]
|
| 89 |
+
INFO:root:Average loss=0.029642372965812683
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1690 | LCC = 0.9703 | SRCC = 0.9697 ]
|
| 92 |
+
INFO:root:Average loss=0.029086005866527556
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1586 | LCC = 0.9709 | SRCC = 0.9698 ]
|
| 95 |
+
INFO:root:Average loss=0.02773505213856697
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1517 | LCC = 0.9713 | SRCC = 0.9701 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1552 | LCC = 0.9730 | SRCC = 0.9730 ]
|
results_tencent/w2v2_xlsr_1b/layer36/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_1b/layer36/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2a1daa260328be8ea79aa3878e76e9315ed54ea79c913a35085a7d1693fae976
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer36/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:198775b37aaf963a4c02686b1ab9cdbe38bcc96c40b482744bdd1855111404dc
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer36/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.4184150598049164
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2139 | LCC = 0.9482 | SRCC = 0.9464 ]
|
| 8 |
+
INFO:root:Average loss=0.14041236561536788
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.1773 | LCC = 0.9571 | SRCC = 0.9560 ]
|
| 11 |
+
INFO:root:Average loss=0.11671010273694991
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1658 | LCC = 0.9622 | SRCC = 0.9618 ]
|
| 14 |
+
INFO:root:Average loss=0.10370067405700684
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1990 | LCC = 0.9648 | SRCC = 0.9637 ]
|
| 17 |
+
INFO:root:Average loss=0.09313969212770462
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2338 | LCC = 0.9672 | SRCC = 0.9661 ]
|
| 20 |
+
INFO:root:Average loss=0.0937761844098568
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1695 | LCC = 0.9682 | SRCC = 0.9672 ]
|
| 23 |
+
INFO:root:Average loss=0.07943975445628167
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2063 | LCC = 0.9687 | SRCC = 0.9680 ]
|
| 26 |
+
INFO:root:Average loss=0.07610993379354478
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1912 | LCC = 0.9695 | SRCC = 0.9686 ]
|
| 29 |
+
INFO:root:Average loss=0.07089036694169044
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1854 | LCC = 0.9697 | SRCC = 0.9689 ]
|
| 32 |
+
INFO:root:Average loss=0.06716691708564758
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1834 | LCC = 0.9697 | SRCC = 0.9690 ]
|
| 35 |
+
INFO:root:Average loss=0.06476069831848144
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2299 | LCC = 0.9705 | SRCC = 0.9693 ]
|
| 38 |
+
INFO:root:Average loss=0.060296593576669694
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1845 | LCC = 0.9701 | SRCC = 0.9694 ]
|
| 41 |
+
INFO:root:Average loss=0.056426486283540724
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2203 | LCC = 0.9705 | SRCC = 0.9697 ]
|
| 44 |
+
INFO:root:Average loss=0.05369547292590141
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1798 | LCC = 0.9708 | SRCC = 0.9698 ]
|
| 47 |
+
INFO:root:Average loss=0.05339738672971726
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1827 | LCC = 0.9716 | SRCC = 0.9705 ]
|
| 50 |
+
INFO:root:Average loss=0.05019618168473244
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1734 | LCC = 0.9702 | SRCC = 0.9689 ]
|
| 53 |
+
INFO:root:Average loss=0.047284602239727976
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1486 | LCC = 0.9685 | SRCC = 0.9694 ]
|
| 56 |
+
INFO:root:Average loss=0.04673340328037739
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1482 | LCC = 0.9703 | SRCC = 0.9696 ]
|
| 59 |
+
INFO:root:Average loss=0.043822299867868425
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1762 | LCC = 0.9699 | SRCC = 0.9686 ]
|
| 62 |
+
INFO:root:Average loss=0.04166319189965725
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1655 | LCC = 0.9700 | SRCC = 0.9696 ]
|
| 65 |
+
INFO:root:Average loss=0.0402757787257433
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1405 | LCC = 0.9697 | SRCC = 0.9692 ]
|
| 68 |
+
INFO:root:Average loss=0.04029739107191563
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1739 | LCC = 0.9710 | SRCC = 0.9700 ]
|
| 71 |
+
INFO:root:Average loss=0.03568760584294796
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1773 | LCC = 0.9706 | SRCC = 0.9697 ]
|
| 74 |
+
INFO:root:Average loss=0.035622006043791773
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1587 | LCC = 0.9711 | SRCC = 0.9699 ]
|
| 77 |
+
INFO:root:Average loss=0.03358796517550945
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1808 | LCC = 0.9689 | SRCC = 0.9685 ]
|
| 80 |
+
INFO:root:Average loss=0.034545952156186104
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1915 | LCC = 0.9714 | SRCC = 0.9703 ]
|
| 83 |
+
INFO:root:Average loss=0.034197140738368034
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1638 | LCC = 0.9717 | SRCC = 0.9703 ]
|
| 86 |
+
INFO:root:Average loss=0.029682789251208305
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1876 | LCC = 0.9709 | SRCC = 0.9700 ]
|
| 89 |
+
INFO:root:Average loss=0.030314480409026148
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1664 | LCC = 0.9712 | SRCC = 0.9700 ]
|
| 92 |
+
INFO:root:Average loss=0.029289673045277596
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1840 | LCC = 0.9707 | SRCC = 0.9695 ]
|
| 95 |
+
INFO:root:Average loss=0.028404358118772507
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1673 | LCC = 0.9722 | SRCC = 0.9707 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1632 | LCC = 0.9729 | SRCC = 0.9724 ]
|
results_tencent/w2v2_xlsr_1b/layer36/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_1b/layer36/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d597d38f707dae6769e94de7e0e27f302bee8ca5363749ca5d635305208af238
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer36/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:10027e9d8351b24020792ccfe100ddc34e9fb239262aa83b05002e1790cd383a
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer36/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.3466171278953552
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2414 | LCC = 0.9474 | SRCC = 0.9473 ]
|
| 8 |
+
INFO:root:Average loss=0.14407474410533905
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2357 | LCC = 0.9567 | SRCC = 0.9586 ]
|
| 11 |
+
INFO:root:Average loss=0.12045472604036331
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1806 | LCC = 0.9602 | SRCC = 0.9620 ]
|
| 14 |
+
INFO:root:Average loss=0.10376573845744133
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2069 | LCC = 0.9658 | SRCC = 0.9656 ]
|
| 17 |
+
INFO:root:Average loss=0.09214080440998078
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1736 | LCC = 0.9660 | SRCC = 0.9656 ]
|
| 20 |
+
INFO:root:Average loss=0.08704169976711273
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2007 | LCC = 0.9672 | SRCC = 0.9676 ]
|
| 23 |
+
INFO:root:Average loss=0.08388708880543709
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1643 | LCC = 0.9688 | SRCC = 0.9686 ]
|
| 26 |
+
INFO:root:Average loss=0.07602933475375176
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1791 | LCC = 0.9682 | SRCC = 0.9677 ]
|
| 29 |
+
INFO:root:Average loss=0.06806914582848549
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1524 | LCC = 0.9688 | SRCC = 0.9690 ]
|
| 32 |
+
INFO:root:Average loss=0.0666362652182579
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1515 | LCC = 0.9693 | SRCC = 0.9698 ]
|
| 35 |
+
INFO:root:Average loss=0.06287260290980339
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1546 | LCC = 0.9701 | SRCC = 0.9695 ]
|
| 38 |
+
INFO:root:Average loss=0.059920053124427795
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1474 | LCC = 0.9698 | SRCC = 0.9701 ]
|
| 41 |
+
INFO:root:Average loss=0.05628760279715061
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1654 | LCC = 0.9707 | SRCC = 0.9703 ]
|
| 44 |
+
INFO:root:Average loss=0.05284775601327419
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1566 | LCC = 0.9706 | SRCC = 0.9691 ]
|
| 47 |
+
INFO:root:Average loss=0.05544387456774712
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1744 | LCC = 0.9709 | SRCC = 0.9700 ]
|
| 50 |
+
INFO:root:Average loss=0.04981437346339226
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1670 | LCC = 0.9708 | SRCC = 0.9698 ]
|
| 53 |
+
INFO:root:Average loss=0.04602447083592415
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1512 | LCC = 0.9711 | SRCC = 0.9700 ]
|
| 56 |
+
INFO:root:Average loss=0.04670155562460423
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1457 | LCC = 0.9710 | SRCC = 0.9705 ]
|
| 59 |
+
INFO:root:Average loss=0.042255980134010315
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1756 | LCC = 0.9702 | SRCC = 0.9698 ]
|
| 62 |
+
INFO:root:Average loss=0.042576082974672316
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1703 | LCC = 0.9709 | SRCC = 0.9700 ]
|
| 65 |
+
INFO:root:Average loss=0.040510559529066084
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1662 | LCC = 0.9713 | SRCC = 0.9704 ]
|
| 68 |
+
INFO:root:Average loss=0.04114079168438912
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1631 | LCC = 0.9718 | SRCC = 0.9708 ]
|
| 71 |
+
INFO:root:Average loss=0.03683482539653778
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1593 | LCC = 0.9716 | SRCC = 0.9704 ]
|
| 74 |
+
INFO:root:Average loss=0.03525456528365612
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1514 | LCC = 0.9714 | SRCC = 0.9707 ]
|
| 77 |
+
INFO:root:Average loss=0.033437778785824775
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1437 | LCC = 0.9712 | SRCC = 0.9703 ]
|
| 80 |
+
INFO:root:Average loss=0.03135680602490902
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1862 | LCC = 0.9706 | SRCC = 0.9702 ]
|
| 83 |
+
INFO:root:Average loss=0.031429137006402014
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1700 | LCC = 0.9715 | SRCC = 0.9704 ]
|
| 86 |
+
INFO:root:Average loss=0.031810793444514276
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1778 | LCC = 0.9701 | SRCC = 0.9700 ]
|
| 89 |
+
INFO:root:Average loss=0.0291527586132288
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1456 | LCC = 0.9713 | SRCC = 0.9701 ]
|
| 92 |
+
INFO:root:Average loss=0.02701096299290657
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1556 | LCC = 0.9709 | SRCC = 0.9697 ]
|
| 95 |
+
INFO:root:Average loss=0.027985470943152905
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1645 | LCC = 0.9710 | SRCC = 0.9702 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1614 | LCC = 0.9726 | SRCC = 0.9724 ]
|
results_tencent/w2v2_xlsr_1b/layer38/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_1b/layer38/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:59928d25e2a3be9bf81fd1236b3641005b5dd820dccb03732cfbc045f300f84d
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer38/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ced1892b6897f7d7e1b781a22ada4790568512a16c2293465bca3e6bd9201d66
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer38/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.3558842204809189
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2582 | LCC = 0.9480 | SRCC = 0.9488 ]
|
| 8 |
+
INFO:root:Average loss=0.14679123330116273
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2523 | LCC = 0.9563 | SRCC = 0.9584 ]
|
| 11 |
+
INFO:root:Average loss=0.11954030108451844
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1919 | LCC = 0.9626 | SRCC = 0.9623 ]
|
| 14 |
+
INFO:root:Average loss=0.10363795626163483
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1817 | LCC = 0.9649 | SRCC = 0.9646 ]
|
| 17 |
+
INFO:root:Average loss=0.09424833685159684
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1755 | LCC = 0.9645 | SRCC = 0.9636 ]
|
| 20 |
+
INFO:root:Average loss=0.08609261974692345
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2040 | LCC = 0.9672 | SRCC = 0.9670 ]
|
| 23 |
+
INFO:root:Average loss=0.07944248157739639
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1761 | LCC = 0.9680 | SRCC = 0.9665 ]
|
| 26 |
+
INFO:root:Average loss=0.07284740918874741
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1894 | LCC = 0.9683 | SRCC = 0.9677 ]
|
| 29 |
+
INFO:root:Average loss=0.07072249266505241
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1508 | LCC = 0.9686 | SRCC = 0.9686 ]
|
| 32 |
+
INFO:root:Average loss=0.06409601128101348
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2092 | LCC = 0.9695 | SRCC = 0.9692 ]
|
| 35 |
+
INFO:root:Average loss=0.06285435909032822
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1575 | LCC = 0.9693 | SRCC = 0.9693 ]
|
| 38 |
+
INFO:root:Average loss=0.061157888621091845
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1841 | LCC = 0.9699 | SRCC = 0.9692 ]
|
| 41 |
+
INFO:root:Average loss=0.05656137952208519
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1888 | LCC = 0.9706 | SRCC = 0.9696 ]
|
| 44 |
+
INFO:root:Average loss=0.058424759060144424
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1548 | LCC = 0.9707 | SRCC = 0.9702 ]
|
| 47 |
+
INFO:root:Average loss=0.05241496953368187
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1711 | LCC = 0.9710 | SRCC = 0.9703 ]
|
| 50 |
+
INFO:root:Average loss=0.049481783553957936
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1939 | LCC = 0.9698 | SRCC = 0.9697 ]
|
| 53 |
+
INFO:root:Average loss=0.048216026574373246
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1747 | LCC = 0.9706 | SRCC = 0.9695 ]
|
| 56 |
+
INFO:root:Average loss=0.04518214210867882
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1487 | LCC = 0.9709 | SRCC = 0.9701 ]
|
| 59 |
+
INFO:root:Average loss=0.04323844361305237
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1860 | LCC = 0.9702 | SRCC = 0.9696 ]
|
| 62 |
+
INFO:root:Average loss=0.04365216052532196
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1851 | LCC = 0.9707 | SRCC = 0.9698 ]
|
| 65 |
+
INFO:root:Average loss=0.0375362349152565
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1776 | LCC = 0.9707 | SRCC = 0.9700 ]
|
| 68 |
+
INFO:root:Average loss=0.03631195691227913
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1786 | LCC = 0.9704 | SRCC = 0.9694 ]
|
| 71 |
+
INFO:root:Average loss=0.038143286049365996
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1639 | LCC = 0.9709 | SRCC = 0.9697 ]
|
| 74 |
+
INFO:root:Average loss=0.03450111666321754
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1653 | LCC = 0.9704 | SRCC = 0.9692 ]
|
| 77 |
+
INFO:root:Average loss=0.037062941417098046
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1852 | LCC = 0.9705 | SRCC = 0.9689 ]
|
| 80 |
+
INFO:root:Average loss=0.033990324556827545
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1829 | LCC = 0.9709 | SRCC = 0.9704 ]
|
| 83 |
+
INFO:root:Average loss=0.029204134047031403
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1677 | LCC = 0.9708 | SRCC = 0.9700 ]
|
| 86 |
+
INFO:root:Average loss=0.029512074708938598
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1497 | LCC = 0.9712 | SRCC = 0.9700 ]
|
| 89 |
+
INFO:root:Average loss=0.032467821285128595
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1717 | LCC = 0.9704 | SRCC = 0.9695 ]
|
| 92 |
+
INFO:root:Average loss=0.02685604488104582
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1542 | LCC = 0.9717 | SRCC = 0.9706 ]
|
| 95 |
+
INFO:root:Average loss=0.028194790333509447
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1723 | LCC = 0.9711 | SRCC = 0.9701 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1474 | LCC = 0.9729 | SRCC = 0.9729 ]
|
results_tencent/w2v2_xlsr_1b/layer38/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_1b/layer38/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d2dec282890a96175b1e01b2b2bd475b71821d654a8bec018850fb24b131c066
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer38/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9c4da0081cdcc580481425017d18abae339e09ecc6bdf13b17abb679199cecd8
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer38/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.3856969577074051
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2479 | LCC = 0.9503 | SRCC = 0.9500 ]
|
| 8 |
+
INFO:root:Average loss=0.13848739260435106
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.1794 | LCC = 0.9602 | SRCC = 0.9592 ]
|
| 11 |
+
INFO:root:Average loss=0.10929290664196015
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.1833 | LCC = 0.9641 | SRCC = 0.9630 ]
|
| 14 |
+
INFO:root:Average loss=0.10242943039536476
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.1645 | LCC = 0.9660 | SRCC = 0.9654 ]
|
| 17 |
+
INFO:root:Average loss=0.09316253691911697
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1752 | LCC = 0.9672 | SRCC = 0.9666 ]
|
| 20 |
+
INFO:root:Average loss=0.0799498461484909
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2254 | LCC = 0.9667 | SRCC = 0.9666 ]
|
| 23 |
+
INFO:root:Average loss=0.07767478197813034
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1577 | LCC = 0.9688 | SRCC = 0.9680 ]
|
| 26 |
+
INFO:root:Average loss=0.07336836779117584
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1594 | LCC = 0.9697 | SRCC = 0.9687 ]
|
| 29 |
+
INFO:root:Average loss=0.06617289617657661
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1471 | LCC = 0.9701 | SRCC = 0.9691 ]
|
| 32 |
+
INFO:root:Average loss=0.06345405882596969
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1835 | LCC = 0.9695 | SRCC = 0.9691 ]
|
| 35 |
+
INFO:root:Average loss=0.06454156318306922
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1850 | LCC = 0.9709 | SRCC = 0.9698 ]
|
| 38 |
+
INFO:root:Average loss=0.05887997670471668
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1436 | LCC = 0.9707 | SRCC = 0.9697 ]
|
| 41 |
+
INFO:root:Average loss=0.055705289766192434
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1428 | LCC = 0.9711 | SRCC = 0.9701 ]
|
| 44 |
+
INFO:root:Average loss=0.054259054869413376
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1486 | LCC = 0.9687 | SRCC = 0.9679 ]
|
| 47 |
+
INFO:root:Average loss=0.05171012181043625
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1434 | LCC = 0.9708 | SRCC = 0.9697 ]
|
| 50 |
+
INFO:root:Average loss=0.04836002166569233
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1379 | LCC = 0.9702 | SRCC = 0.9692 ]
|
| 53 |
+
INFO:root:Average loss=0.043975448697805404
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1540 | LCC = 0.9695 | SRCC = 0.9689 ]
|
| 56 |
+
INFO:root:Average loss=0.04366652525961399
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1393 | LCC = 0.9676 | SRCC = 0.9677 ]
|
| 59 |
+
INFO:root:Average loss=0.04300688318908215
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1425 | LCC = 0.9714 | SRCC = 0.9701 ]
|
| 62 |
+
INFO:root:Average loss=0.04092701049149036
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1449 | LCC = 0.9709 | SRCC = 0.9696 ]
|
| 65 |
+
INFO:root:Average loss=0.0406169667840004
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1398 | LCC = 0.9712 | SRCC = 0.9700 ]
|
| 68 |
+
INFO:root:Average loss=0.03672604832053185
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1507 | LCC = 0.9715 | SRCC = 0.9702 ]
|
| 71 |
+
INFO:root:Average loss=0.03712812514603138
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1393 | LCC = 0.9707 | SRCC = 0.9694 ]
|
| 74 |
+
INFO:root:Average loss=0.034318795174360275
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1472 | LCC = 0.9689 | SRCC = 0.9684 ]
|
| 77 |
+
INFO:root:Average loss=0.03358851247280836
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1424 | LCC = 0.9711 | SRCC = 0.9701 ]
|
| 80 |
+
INFO:root:Average loss=0.034150296851992605
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1479 | LCC = 0.9713 | SRCC = 0.9700 ]
|
| 83 |
+
INFO:root:Average loss=0.031058604896068574
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1514 | LCC = 0.9716 | SRCC = 0.9701 ]
|
| 86 |
+
INFO:root:Average loss=0.02965648752450943
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1677 | LCC = 0.9714 | SRCC = 0.9698 ]
|
| 89 |
+
INFO:root:Average loss=0.029592418894171715
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1440 | LCC = 0.9702 | SRCC = 0.9693 ]
|
| 92 |
+
INFO:root:Average loss=0.03005637326091528
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1457 | LCC = 0.9709 | SRCC = 0.9698 ]
|
| 95 |
+
INFO:root:Average loss=0.027972604781389236
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1509 | LCC = 0.9710 | SRCC = 0.9698 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1447 | LCC = 0.9735 | SRCC = 0.9734 ]
|
results_tencent/w2v2_xlsr_1b/layer38/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_1b/layer38/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:58451ff715e8da175bb5def734bb6524daa0a2b33207923442199818a9fb8a45
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer38/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:02995df1b59741c46dda3cd9bf92282b6e621aa291ef3bd66798961efee5de1d
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer38/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.3794033288955688
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.3109 | LCC = 0.9459 | SRCC = 0.9455 ]
|
| 8 |
+
INFO:root:Average loss=0.1416382457613945
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2201 | LCC = 0.9597 | SRCC = 0.9579 ]
|
| 11 |
+
INFO:root:Average loss=0.11196582877635956
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2155 | LCC = 0.9634 | SRCC = 0.9624 ]
|
| 14 |
+
INFO:root:Average loss=0.09794212204217911
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2410 | LCC = 0.9653 | SRCC = 0.9641 ]
|
| 17 |
+
INFO:root:Average loss=0.09182572892308236
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.1583 | LCC = 0.9654 | SRCC = 0.9647 ]
|
| 20 |
+
INFO:root:Average loss=0.0865036818087101
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.1957 | LCC = 0.9677 | SRCC = 0.9661 ]
|
| 23 |
+
INFO:root:Average loss=0.07724099358916282
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1629 | LCC = 0.9681 | SRCC = 0.9669 ]
|
| 26 |
+
INFO:root:Average loss=0.07598049944639206
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1692 | LCC = 0.9670 | SRCC = 0.9665 ]
|
| 29 |
+
INFO:root:Average loss=0.0742725749015808
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2247 | LCC = 0.9690 | SRCC = 0.9682 ]
|
| 32 |
+
INFO:root:Average loss=0.06504936426877976
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1646 | LCC = 0.9698 | SRCC = 0.9687 ]
|
| 35 |
+
INFO:root:Average loss=0.06173207664489746
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1995 | LCC = 0.9701 | SRCC = 0.9691 ]
|
| 38 |
+
INFO:root:Average loss=0.05952079963684082
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1696 | LCC = 0.9701 | SRCC = 0.9691 ]
|
| 41 |
+
INFO:root:Average loss=0.05654339304566383
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1872 | LCC = 0.9689 | SRCC = 0.9683 ]
|
| 44 |
+
INFO:root:Average loss=0.05348136024177074
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2044 | LCC = 0.9694 | SRCC = 0.9690 ]
|
| 47 |
+
INFO:root:Average loss=0.05224865095317364
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1704 | LCC = 0.9705 | SRCC = 0.9694 ]
|
| 50 |
+
INFO:root:Average loss=0.04908099475502968
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2130 | LCC = 0.9697 | SRCC = 0.9691 ]
|
| 53 |
+
INFO:root:Average loss=0.04488840971887112
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1875 | LCC = 0.9696 | SRCC = 0.9690 ]
|
| 56 |
+
INFO:root:Average loss=0.04518809826672077
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1737 | LCC = 0.9697 | SRCC = 0.9681 ]
|
| 59 |
+
INFO:root:Average loss=0.04163896660506725
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1696 | LCC = 0.9705 | SRCC = 0.9690 ]
|
| 62 |
+
INFO:root:Average loss=0.03920104588568211
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1682 | LCC = 0.9708 | SRCC = 0.9697 ]
|
| 65 |
+
INFO:root:Average loss=0.040962110802531246
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1575 | LCC = 0.9701 | SRCC = 0.9690 ]
|
| 68 |
+
INFO:root:Average loss=0.03647107282280922
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1839 | LCC = 0.9709 | SRCC = 0.9696 ]
|
| 71 |
+
INFO:root:Average loss=0.0365851811170578
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1741 | LCC = 0.9705 | SRCC = 0.9689 ]
|
| 74 |
+
INFO:root:Average loss=0.036303778678178786
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1897 | LCC = 0.9703 | SRCC = 0.9688 ]
|
| 77 |
+
INFO:root:Average loss=0.03547481192648411
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1538 | LCC = 0.9692 | SRCC = 0.9681 ]
|
| 80 |
+
INFO:root:Average loss=0.0331257444024086
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1548 | LCC = 0.9703 | SRCC = 0.9689 ]
|
| 83 |
+
INFO:root:Average loss=0.031102280110120773
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1731 | LCC = 0.9708 | SRCC = 0.9693 ]
|
| 86 |
+
INFO:root:Average loss=0.03095718766748905
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1862 | LCC = 0.9702 | SRCC = 0.9690 ]
|
| 89 |
+
INFO:root:Average loss=0.03009469159692526
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1750 | LCC = 0.9702 | SRCC = 0.9685 ]
|
| 92 |
+
INFO:root:Average loss=0.02784664736688137
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1705 | LCC = 0.9708 | SRCC = 0.9692 ]
|
| 95 |
+
INFO:root:Average loss=0.02976624257862568
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1592 | LCC = 0.9712 | SRCC = 0.9698 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1504 | LCC = 0.9738 | SRCC = 0.9734 ]
|
results_tencent/w2v2_xlsr_1b/layer38/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_1b/layer38/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:da611060cbeb0c9bf5b0c4f53131c2d9a0ba4609631ec8f1fe82016d693719e3
|
| 3 |
+
size 4058699
|
results_tencent/w2v2_xlsr_1b/layer38/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:16339bda6cf6b0049d8fa4ee56fc867857126e106323f197385d62ff5f317bc3
|
| 3 |
+
size 4060358
|
results_tencent/w2v2_xlsr_1b/layer38/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.39518193674087526
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.2611 | LCC = 0.9497 | SRCC = 0.9502 ]
|
| 8 |
+
INFO:root:Average loss=0.1414701886177063
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.2556 | LCC = 0.9602 | SRCC = 0.9603 ]
|
| 11 |
+
INFO:root:Average loss=0.1182013139128685
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.2213 | LCC = 0.9631 | SRCC = 0.9625 ]
|
| 14 |
+
INFO:root:Average loss=0.10103740113973618
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.2178 | LCC = 0.9652 | SRCC = 0.9645 ]
|
| 17 |
+
INFO:root:Average loss=0.08940617421269417
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2279 | LCC = 0.9663 | SRCC = 0.9657 ]
|
| 20 |
+
INFO:root:Average loss=0.0874656894505024
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.2144 | LCC = 0.9681 | SRCC = 0.9676 ]
|
| 23 |
+
INFO:root:Average loss=0.07892866724729539
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.1938 | LCC = 0.9685 | SRCC = 0.9680 ]
|
| 26 |
+
INFO:root:Average loss=0.07517336672544479
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.1862 | LCC = 0.9678 | SRCC = 0.9676 ]
|
| 29 |
+
INFO:root:Average loss=0.07207538169622421
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.1944 | LCC = 0.9685 | SRCC = 0.9679 ]
|
| 32 |
+
INFO:root:Average loss=0.06571751376986504
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.1810 | LCC = 0.9694 | SRCC = 0.9682 ]
|
| 35 |
+
INFO:root:Average loss=0.06114544069766998
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.1924 | LCC = 0.9697 | SRCC = 0.9688 ]
|
| 38 |
+
INFO:root:Average loss=0.05673066011071205
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.1548 | LCC = 0.9692 | SRCC = 0.9690 ]
|
| 41 |
+
INFO:root:Average loss=0.05721793439984322
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.1835 | LCC = 0.9699 | SRCC = 0.9693 ]
|
| 44 |
+
INFO:root:Average loss=0.054944389790296554
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.1819 | LCC = 0.9698 | SRCC = 0.9692 ]
|
| 47 |
+
INFO:root:Average loss=0.049349200055003165
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.1681 | LCC = 0.9702 | SRCC = 0.9693 ]
|
| 50 |
+
INFO:root:Average loss=0.04871702240407467
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.1893 | LCC = 0.9701 | SRCC = 0.9696 ]
|
| 53 |
+
INFO:root:Average loss=0.04712915739417076
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.1754 | LCC = 0.9701 | SRCC = 0.9691 ]
|
| 56 |
+
INFO:root:Average loss=0.04491954132914543
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.1957 | LCC = 0.9698 | SRCC = 0.9696 ]
|
| 59 |
+
INFO:root:Average loss=0.04172478725016117
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.1759 | LCC = 0.9706 | SRCC = 0.9696 ]
|
| 62 |
+
INFO:root:Average loss=0.0407626311480999
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.1584 | LCC = 0.9700 | SRCC = 0.9692 ]
|
| 65 |
+
INFO:root:Average loss=0.038834272131323816
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.1638 | LCC = 0.9706 | SRCC = 0.9694 ]
|
| 68 |
+
INFO:root:Average loss=0.037612869411706926
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.1761 | LCC = 0.9707 | SRCC = 0.9697 ]
|
| 71 |
+
INFO:root:Average loss=0.035039506554603576
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.1754 | LCC = 0.9705 | SRCC = 0.9696 ]
|
| 74 |
+
INFO:root:Average loss=0.03711528511345386
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.1859 | LCC = 0.9707 | SRCC = 0.9695 ]
|
| 77 |
+
INFO:root:Average loss=0.03994645991921425
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.1663 | LCC = 0.9704 | SRCC = 0.9690 ]
|
| 80 |
+
INFO:root:Average loss=0.03346462886035442
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.1829 | LCC = 0.9706 | SRCC = 0.9699 ]
|
| 83 |
+
INFO:root:Average loss=0.031186686784029007
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.1576 | LCC = 0.9704 | SRCC = 0.9691 ]
|
| 86 |
+
INFO:root:Average loss=0.03445906959474087
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.1645 | LCC = 0.9709 | SRCC = 0.9696 ]
|
| 89 |
+
INFO:root:Average loss=0.030764516606926917
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.1582 | LCC = 0.9708 | SRCC = 0.9693 ]
|
| 92 |
+
INFO:root:Average loss=0.028681699320673944
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.1540 | LCC = 0.9695 | SRCC = 0.9689 ]
|
| 95 |
+
INFO:root:Average loss=0.02761603620648384
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.1727 | LCC = 0.9702 | SRCC = 0.9688 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.1602 | LCC = 0.9725 | SRCC = 0.9716 ]
|