Upload SSL Layer MOS checkpoints (part 9)
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- results_bvcc/w2v2_large_asr_en/layer10/run2/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer10/run2/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer10/run2/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer10/run3/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer10/run3/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer10/run3/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer10/run3/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer10/run4/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer10/run4/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer10/run4/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer10/run4/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer12/run0/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer12/run0/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run0/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run0/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer12/run1/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer12/run1/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run1/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run1/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer12/run2/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer12/run2/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run2/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run2/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer12/run3/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer12/run3/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run3/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run3/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer12/run4/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer12/run4/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run4/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer12/run4/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer14/run0/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer14/run0/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run0/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run0/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer14/run1/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer14/run1/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run1/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run1/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer14/run2/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer14/run2/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run2/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run2/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer14/run3/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer14/run3/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run3/model_best.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run3/train.log +99 -0
- results_bvcc/w2v2_large_asr_en/layer14/run4/config.gin +31 -0
- results_bvcc/w2v2_large_asr_en/layer14/run4/model.pt +3 -0
- results_bvcc/w2v2_large_asr_en/layer14/run4/model_best.pt +3 -0
results_bvcc/w2v2_large_asr_en/layer10/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:17c954b27d304f577e9fd998036c8afa372b2fb1bb1ecb167af33267fe87ebd6
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer10/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b982e79fea04f796dbed29dabf4b3392178d6f9164e38998293ad06c9387d387
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer10/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.6746184119047263
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4778 | LCC = 0.7159 | SRCC = 0.7216 ]
|
| 8 |
+
INFO:root:Average loss=0.3387069020133752
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3338 | LCC = 0.8053 | SRCC = 0.8036 ]
|
| 11 |
+
INFO:root:Average loss=0.2847767711067811
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3085 | LCC = 0.8174 | SRCC = 0.8155 ]
|
| 14 |
+
INFO:root:Average loss=0.26873792765232235
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3536 | LCC = 0.8174 | SRCC = 0.8175 ]
|
| 17 |
+
INFO:root:Average loss=0.23792060521932748
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3200 | LCC = 0.8162 | SRCC = 0.8128 ]
|
| 20 |
+
INFO:root:Average loss=0.2266076065790959
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3025 | LCC = 0.8133 | SRCC = 0.8105 ]
|
| 23 |
+
INFO:root:Average loss=0.21253541111946106
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3221 | LCC = 0.8146 | SRCC = 0.8157 ]
|
| 26 |
+
INFO:root:Average loss=0.19001203135420114
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3453 | LCC = 0.8117 | SRCC = 0.8131 ]
|
| 29 |
+
INFO:root:Average loss=0.17686052477130523
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2949 | LCC = 0.8136 | SRCC = 0.8142 ]
|
| 32 |
+
INFO:root:Average loss=0.16311080323962066
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3003 | LCC = 0.8100 | SRCC = 0.8084 ]
|
| 35 |
+
INFO:root:Average loss=0.1606394347663109
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3156 | LCC = 0.8080 | SRCC = 0.8061 ]
|
| 38 |
+
INFO:root:Average loss=0.14482228400615546
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3110 | LCC = 0.8137 | SRCC = 0.8113 ]
|
| 41 |
+
INFO:root:Average loss=0.14077927630681258
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2824 | LCC = 0.8181 | SRCC = 0.8155 ]
|
| 44 |
+
INFO:root:Average loss=0.1233119808901579
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2920 | LCC = 0.8107 | SRCC = 0.8101 ]
|
| 47 |
+
INFO:root:Average loss=0.12160828308417247
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2835 | LCC = 0.8124 | SRCC = 0.8111 ]
|
| 50 |
+
INFO:root:Average loss=0.11040873868534198
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2900 | LCC = 0.8126 | SRCC = 0.8101 ]
|
| 53 |
+
INFO:root:Average loss=0.10938038973089977
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2914 | LCC = 0.8137 | SRCC = 0.8119 ]
|
| 56 |
+
INFO:root:Average loss=0.10012815907024421
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3084 | LCC = 0.8083 | SRCC = 0.8057 ]
|
| 59 |
+
INFO:root:Average loss=0.10388120755744286
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2943 | LCC = 0.8107 | SRCC = 0.8089 ]
|
| 62 |
+
INFO:root:Average loss=0.08978882174079235
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3115 | LCC = 0.8074 | SRCC = 0.8049 ]
|
| 65 |
+
INFO:root:Average loss=0.08635260518162678
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3136 | LCC = 0.8057 | SRCC = 0.8069 ]
|
| 68 |
+
INFO:root:Average loss=0.07857741916982028
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2885 | LCC = 0.8124 | SRCC = 0.8121 ]
|
| 71 |
+
INFO:root:Average loss=0.08343375440782462
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3734 | LCC = 0.8007 | SRCC = 0.8059 ]
|
| 74 |
+
INFO:root:Average loss=0.08682268590499194
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3107 | LCC = 0.8072 | SRCC = 0.8076 ]
|
| 77 |
+
INFO:root:Average loss=0.07026724560329548
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2972 | LCC = 0.8077 | SRCC = 0.8065 ]
|
| 80 |
+
INFO:root:Average loss=0.0699538896099115
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3158 | LCC = 0.8078 | SRCC = 0.8084 ]
|
| 83 |
+
INFO:root:Average loss=0.0652948336628003
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2908 | LCC = 0.8137 | SRCC = 0.8124 ]
|
| 86 |
+
INFO:root:Average loss=0.06554413371934341
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2928 | LCC = 0.8084 | SRCC = 0.8075 ]
|
| 89 |
+
INFO:root:Average loss=0.06051074364819588
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3119 | LCC = 0.8020 | SRCC = 0.8009 ]
|
| 92 |
+
INFO:root:Average loss=0.0663526530067126
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3253 | LCC = 0.8010 | SRCC = 0.8017 ]
|
| 95 |
+
INFO:root:Average loss=0.06058197826720201
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3325 | LCC = 0.8038 | SRCC = 0.8048 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2816 | LCC = 0.8313 | SRCC = 0.8339 ]
|
results_bvcc/w2v2_large_asr_en/layer10/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer10/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7a9677dbfb8f2aad402e8378478a102d4599225892a987b760e678b5ed4ccc42
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer10/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bef1cdb12deabd38fd53fc24ae27e1f98b2f9318a55b33d24b2f0916d0fb429c
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer10/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5988704699736375
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4256 | LCC = 0.7561 | SRCC = 0.7585 ]
|
| 8 |
+
INFO:root:Average loss=0.3351873986614056
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3306 | LCC = 0.8009 | SRCC = 0.8010 ]
|
| 11 |
+
INFO:root:Average loss=0.2886387335184293
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3175 | LCC = 0.8080 | SRCC = 0.8062 ]
|
| 14 |
+
INFO:root:Average loss=0.2622057657975417
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3165 | LCC = 0.8089 | SRCC = 0.8082 ]
|
| 17 |
+
INFO:root:Average loss=0.2366573931888128
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.2980 | LCC = 0.8144 | SRCC = 0.8138 ]
|
| 20 |
+
INFO:root:Average loss=0.22310085451373687
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3147 | LCC = 0.8087 | SRCC = 0.8086 ]
|
| 23 |
+
INFO:root:Average loss=0.23775893927384645
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2993 | LCC = 0.8173 | SRCC = 0.8149 ]
|
| 26 |
+
INFO:root:Average loss=0.19699565645975944
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2875 | LCC = 0.8165 | SRCC = 0.8149 ]
|
| 29 |
+
INFO:root:Average loss=0.17750538675448832
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3055 | LCC = 0.8035 | SRCC = 0.8018 ]
|
| 32 |
+
INFO:root:Average loss=0.18402894739157113
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2866 | LCC = 0.8115 | SRCC = 0.8086 ]
|
| 35 |
+
INFO:root:Average loss=0.15437620715835157
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3052 | LCC = 0.8091 | SRCC = 0.8077 ]
|
| 38 |
+
INFO:root:Average loss=0.15082182897589144
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3023 | LCC = 0.8042 | SRCC = 0.8035 ]
|
| 41 |
+
INFO:root:Average loss=0.14588640639797235
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3007 | LCC = 0.8087 | SRCC = 0.8076 ]
|
| 44 |
+
INFO:root:Average loss=0.13715299237997103
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3086 | LCC = 0.8106 | SRCC = 0.8090 ]
|
| 47 |
+
INFO:root:Average loss=0.12994531336694193
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2964 | LCC = 0.8083 | SRCC = 0.8081 ]
|
| 50 |
+
INFO:root:Average loss=0.11570002855016635
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2924 | LCC = 0.8070 | SRCC = 0.8052 ]
|
| 53 |
+
INFO:root:Average loss=0.10925124208323467
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2987 | LCC = 0.8022 | SRCC = 0.8017 ]
|
| 56 |
+
INFO:root:Average loss=0.10947629188497861
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.2971 | LCC = 0.8047 | SRCC = 0.8042 ]
|
| 59 |
+
INFO:root:Average loss=0.11201404579556905
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3016 | LCC = 0.8077 | SRCC = 0.8076 ]
|
| 62 |
+
INFO:root:Average loss=0.09224934384035759
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2914 | LCC = 0.8080 | SRCC = 0.8071 ]
|
| 65 |
+
INFO:root:Average loss=0.08938650796428704
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3218 | LCC = 0.8049 | SRCC = 0.8047 ]
|
| 68 |
+
INFO:root:Average loss=0.09217797339153595
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3163 | LCC = 0.8044 | SRCC = 0.8039 ]
|
| 71 |
+
INFO:root:Average loss=0.08819882896466133
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3013 | LCC = 0.8039 | SRCC = 0.8029 ]
|
| 74 |
+
INFO:root:Average loss=0.07946886528187837
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2983 | LCC = 0.8056 | SRCC = 0.8044 ]
|
| 77 |
+
INFO:root:Average loss=0.07416030201010215
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3147 | LCC = 0.8069 | SRCC = 0.8071 ]
|
| 80 |
+
INFO:root:Average loss=0.07000843511942105
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3102 | LCC = 0.8058 | SRCC = 0.8061 ]
|
| 83 |
+
INFO:root:Average loss=0.07103851313392322
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3075 | LCC = 0.8059 | SRCC = 0.8064 ]
|
| 86 |
+
INFO:root:Average loss=0.07397129558599912
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3003 | LCC = 0.8026 | SRCC = 0.8048 ]
|
| 89 |
+
INFO:root:Average loss=0.0688283399034005
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2973 | LCC = 0.8031 | SRCC = 0.8037 ]
|
| 92 |
+
INFO:root:Average loss=0.06074103789451795
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3053 | LCC = 0.8039 | SRCC = 0.8037 ]
|
| 95 |
+
INFO:root:Average loss=0.05725609396512692
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3043 | LCC = 0.8031 | SRCC = 0.8028 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2946 | LCC = 0.8287 | SRCC = 0.8330 ]
|
results_bvcc/w2v2_large_asr_en/layer10/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer10/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:43b280e478e0984619e04a636aa21c3ca58b59766f3a1b907a3e1a28a2bf5628
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer10/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0908d121f460742f853bf06173266c352b2723d6d316379565c5b860f3781a2
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_asr_en/layer10/run4/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.6203618072546445
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4357 | LCC = 0.7424 | SRCC = 0.7445 ]
|
| 8 |
+
INFO:root:Average loss=0.32978219195054126
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3305 | LCC = 0.8015 | SRCC = 0.8003 ]
|
| 11 |
+
INFO:root:Average loss=0.2830068669640101
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3209 | LCC = 0.8173 | SRCC = 0.8138 ]
|
| 14 |
+
INFO:root:Average loss=0.2473831717402507
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3117 | LCC = 0.8149 | SRCC = 0.8109 ]
|
| 17 |
+
INFO:root:Average loss=0.2436703187532914
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3327 | LCC = 0.8152 | SRCC = 0.8133 ]
|
| 20 |
+
INFO:root:Average loss=0.22828983591917235
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3274 | LCC = 0.8206 | SRCC = 0.8169 ]
|
| 23 |
+
INFO:root:Average loss=0.19720400258516654
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2941 | LCC = 0.8195 | SRCC = 0.8157 ]
|
| 26 |
+
INFO:root:Average loss=0.199496636310449
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2920 | LCC = 0.8163 | SRCC = 0.8138 ]
|
| 29 |
+
INFO:root:Average loss=0.17800079887876144
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2984 | LCC = 0.8118 | SRCC = 0.8102 ]
|
| 32 |
+
INFO:root:Average loss=0.16389273231228194
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2954 | LCC = 0.8107 | SRCC = 0.8117 ]
|
| 35 |
+
INFO:root:Average loss=0.15842864261223719
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3438 | LCC = 0.8145 | SRCC = 0.8152 ]
|
| 38 |
+
INFO:root:Average loss=0.1473681323039226
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3195 | LCC = 0.8119 | SRCC = 0.8122 ]
|
| 41 |
+
INFO:root:Average loss=0.14767956275206345
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2986 | LCC = 0.8129 | SRCC = 0.8106 ]
|
| 44 |
+
INFO:root:Average loss=0.12661838818054932
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3090 | LCC = 0.8052 | SRCC = 0.8046 ]
|
| 47 |
+
INFO:root:Average loss=0.12022751426467529
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2973 | LCC = 0.8087 | SRCC = 0.8068 ]
|
| 50 |
+
INFO:root:Average loss=0.11529687218941174
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2894 | LCC = 0.8115 | SRCC = 0.8105 ]
|
| 53 |
+
INFO:root:Average loss=0.11008151047504865
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3617 | LCC = 0.8013 | SRCC = 0.8013 ]
|
| 56 |
+
INFO:root:Average loss=0.10258800918475175
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3020 | LCC = 0.8053 | SRCC = 0.8036 ]
|
| 59 |
+
INFO:root:Average loss=0.09492622760052864
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2955 | LCC = 0.8096 | SRCC = 0.8073 ]
|
| 62 |
+
INFO:root:Average loss=0.09255884143595512
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3150 | LCC = 0.8041 | SRCC = 0.8029 ]
|
| 65 |
+
INFO:root:Average loss=0.0843467849951524
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2996 | LCC = 0.8023 | SRCC = 0.8005 ]
|
| 68 |
+
INFO:root:Average loss=0.0875364528157008
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3082 | LCC = 0.8034 | SRCC = 0.8041 ]
|
| 71 |
+
INFO:root:Average loss=0.08333408980606458
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3099 | LCC = 0.8018 | SRCC = 0.8019 ]
|
| 74 |
+
INFO:root:Average loss=0.0776115395128727
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3091 | LCC = 0.7993 | SRCC = 0.7975 ]
|
| 77 |
+
INFO:root:Average loss=0.07417628866357681
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3151 | LCC = 0.7950 | SRCC = 0.7930 ]
|
| 80 |
+
INFO:root:Average loss=0.07346136371294658
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3040 | LCC = 0.8005 | SRCC = 0.7995 ]
|
| 83 |
+
INFO:root:Average loss=0.06646742208454853
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3142 | LCC = 0.8004 | SRCC = 0.7992 ]
|
| 86 |
+
INFO:root:Average loss=0.06499218501341648
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3218 | LCC = 0.8002 | SRCC = 0.8004 ]
|
| 89 |
+
INFO:root:Average loss=0.06706518684633267
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3453 | LCC = 0.7962 | SRCC = 0.7969 ]
|
| 92 |
+
INFO:root:Average loss=0.06496112478467134
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3409 | LCC = 0.8005 | SRCC = 0.8014 ]
|
| 95 |
+
INFO:root:Average loss=0.05908839113246172
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3119 | LCC = 0.7975 | SRCC = 0.7962 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3395 | LCC = 0.8158 | SRCC = 0.8257 ]
|
results_bvcc/w2v2_large_asr_en/layer12/run0/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer12/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9627174152b51416edf38ab0aaeb16341f92d0a3eae0b4a5fea045918080aa2
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer12/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b3b06a62fff74787b637a935ffe19ad02e0bfb8a2f69a6353632c73c4a258b3
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer12/run0/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5967773099740347
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5083 | LCC = 0.7323 | SRCC = 0.7392 ]
|
| 8 |
+
INFO:root:Average loss=0.34689008731108445
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3440 | LCC = 0.7944 | SRCC = 0.7930 ]
|
| 11 |
+
INFO:root:Average loss=0.3308933778451039
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3955 | LCC = 0.7990 | SRCC = 0.8004 ]
|
| 14 |
+
INFO:root:Average loss=0.2722178880984967
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3819 | LCC = 0.8036 | SRCC = 0.8026 ]
|
| 17 |
+
INFO:root:Average loss=0.2590073903019612
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3180 | LCC = 0.8129 | SRCC = 0.8098 ]
|
| 20 |
+
INFO:root:Average loss=0.23053675374159446
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3261 | LCC = 0.8118 | SRCC = 0.8087 ]
|
| 23 |
+
INFO:root:Average loss=0.21345278983696914
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3180 | LCC = 0.8190 | SRCC = 0.8155 ]
|
| 26 |
+
INFO:root:Average loss=0.20548403683381203
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3376 | LCC = 0.8126 | SRCC = 0.8108 ]
|
| 29 |
+
INFO:root:Average loss=0.20480541846691033
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3139 | LCC = 0.8094 | SRCC = 0.8083 ]
|
| 32 |
+
INFO:root:Average loss=0.17525660494963327
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2926 | LCC = 0.8121 | SRCC = 0.8099 ]
|
| 35 |
+
INFO:root:Average loss=0.17095294269995812
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2990 | LCC = 0.8079 | SRCC = 0.8063 ]
|
| 38 |
+
INFO:root:Average loss=0.15322914127355966
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3106 | LCC = 0.8120 | SRCC = 0.8091 ]
|
| 41 |
+
INFO:root:Average loss=0.14001862017007974
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3499 | LCC = 0.8112 | SRCC = 0.8095 ]
|
| 44 |
+
INFO:root:Average loss=0.15172623604153976
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3013 | LCC = 0.8063 | SRCC = 0.8045 ]
|
| 47 |
+
INFO:root:Average loss=0.14074587439879394
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3060 | LCC = 0.8110 | SRCC = 0.8091 ]
|
| 50 |
+
INFO:root:Average loss=0.12415926702893697
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2993 | LCC = 0.8074 | SRCC = 0.8067 ]
|
| 53 |
+
INFO:root:Average loss=0.1146744348299809
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3094 | LCC = 0.8033 | SRCC = 0.8015 ]
|
| 56 |
+
INFO:root:Average loss=0.11406851951510479
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3587 | LCC = 0.7996 | SRCC = 0.7986 ]
|
| 59 |
+
INFO:root:Average loss=0.10659033136490063
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3100 | LCC = 0.8036 | SRCC = 0.8026 ]
|
| 62 |
+
INFO:root:Average loss=0.09453476110521035
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3329 | LCC = 0.8067 | SRCC = 0.8045 ]
|
| 65 |
+
INFO:root:Average loss=0.08874362950714734
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2957 | LCC = 0.8076 | SRCC = 0.8051 ]
|
| 68 |
+
INFO:root:Average loss=0.09242981037077232
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3054 | LCC = 0.8065 | SRCC = 0.8043 ]
|
| 71 |
+
INFO:root:Average loss=0.07949087367607997
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3520 | LCC = 0.8035 | SRCC = 0.8004 ]
|
| 74 |
+
INFO:root:Average loss=0.07928717561448231
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2984 | LCC = 0.8094 | SRCC = 0.8073 ]
|
| 77 |
+
INFO:root:Average loss=0.08720068447291851
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3126 | LCC = 0.8090 | SRCC = 0.8061 ]
|
| 80 |
+
INFO:root:Average loss=0.07371282224089672
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3219 | LCC = 0.8051 | SRCC = 0.8041 ]
|
| 83 |
+
INFO:root:Average loss=0.07177434422266789
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2913 | LCC = 0.8100 | SRCC = 0.8077 ]
|
| 86 |
+
INFO:root:Average loss=0.06913609372881743
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3401 | LCC = 0.8019 | SRCC = 0.7991 ]
|
| 89 |
+
INFO:root:Average loss=0.0761683588513197
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3065 | LCC = 0.8055 | SRCC = 0.8031 ]
|
| 92 |
+
INFO:root:Average loss=0.0654251968058256
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3148 | LCC = 0.8042 | SRCC = 0.8018 ]
|
| 95 |
+
INFO:root:Average loss=0.06390051720425105
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3053 | LCC = 0.8099 | SRCC = 0.8086 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2847 | LCC = 0.8197 | SRCC = 0.8195 ]
|
results_bvcc/w2v2_large_asr_en/layer12/run1/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer12/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bf84ab06f58f2a54803a6a19b3781544137bdae1ecba533d2c3d8b94ea169253
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer12/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3196ed7c16941f06cf9acdbc5fae9e8942902b9695a99be04a78db14bdab4d47
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer12/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.6111231052722687
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4574 | LCC = 0.7163 | SRCC = 0.7186 ]
|
| 8 |
+
INFO:root:Average loss=0.3580839347380858
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.4009 | LCC = 0.7808 | SRCC = 0.7832 ]
|
| 11 |
+
INFO:root:Average loss=0.30512720556595385
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3525 | LCC = 0.8025 | SRCC = 0.8021 ]
|
| 14 |
+
INFO:root:Average loss=0.2711172887148001
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3829 | LCC = 0.8011 | SRCC = 0.8019 ]
|
| 17 |
+
INFO:root:Average loss=0.2620871812105179
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.4072 | LCC = 0.8052 | SRCC = 0.8094 ]
|
| 20 |
+
INFO:root:Average loss=0.2431554207817102
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3058 | LCC = 0.8136 | SRCC = 0.8097 ]
|
| 23 |
+
INFO:root:Average loss=0.2137223243331298
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3034 | LCC = 0.8146 | SRCC = 0.8108 ]
|
| 26 |
+
INFO:root:Average loss=0.2017128905042624
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3873 | LCC = 0.8064 | SRCC = 0.8097 ]
|
| 29 |
+
INFO:root:Average loss=0.18478004701244524
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2929 | LCC = 0.8128 | SRCC = 0.8109 ]
|
| 32 |
+
INFO:root:Average loss=0.17394910303828043
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2902 | LCC = 0.8135 | SRCC = 0.8111 ]
|
| 35 |
+
INFO:root:Average loss=0.1696313469646833
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3878 | LCC = 0.8061 | SRCC = 0.8076 ]
|
| 38 |
+
INFO:root:Average loss=0.17139079574591073
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3245 | LCC = 0.8067 | SRCC = 0.8059 ]
|
| 41 |
+
INFO:root:Average loss=0.13853010191367224
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3798 | LCC = 0.7954 | SRCC = 0.7975 ]
|
| 44 |
+
INFO:root:Average loss=0.13903944729230341
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3053 | LCC = 0.8020 | SRCC = 0.7972 ]
|
| 47 |
+
INFO:root:Average loss=0.12900117956675017
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2957 | LCC = 0.8049 | SRCC = 0.8025 ]
|
| 50 |
+
INFO:root:Average loss=0.12103551416060863
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3212 | LCC = 0.8053 | SRCC = 0.8034 ]
|
| 53 |
+
INFO:root:Average loss=0.11534078113543682
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3003 | LCC = 0.8066 | SRCC = 0.8030 ]
|
| 56 |
+
INFO:root:Average loss=0.10717813364970379
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3113 | LCC = 0.8017 | SRCC = 0.7972 ]
|
| 59 |
+
INFO:root:Average loss=0.10493040657960452
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2933 | LCC = 0.8079 | SRCC = 0.8048 ]
|
| 62 |
+
INFO:root:Average loss=0.0925721837541996
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3083 | LCC = 0.8038 | SRCC = 0.7998 ]
|
| 65 |
+
INFO:root:Average loss=0.08859112156698337
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3068 | LCC = 0.7973 | SRCC = 0.7938 ]
|
| 68 |
+
INFO:root:Average loss=0.09077216226321
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3093 | LCC = 0.8010 | SRCC = 0.7955 ]
|
| 71 |
+
INFO:root:Average loss=0.08821797834183925
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3173 | LCC = 0.8019 | SRCC = 0.8002 ]
|
| 74 |
+
INFO:root:Average loss=0.07885123287829068
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3267 | LCC = 0.7952 | SRCC = 0.7919 ]
|
| 77 |
+
INFO:root:Average loss=0.07573024904689728
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3180 | LCC = 0.7988 | SRCC = 0.7968 ]
|
| 80 |
+
INFO:root:Average loss=0.06840286609262992
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3259 | LCC = 0.7977 | SRCC = 0.7973 ]
|
| 83 |
+
INFO:root:Average loss=0.06431310022106537
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3008 | LCC = 0.8013 | SRCC = 0.7996 ]
|
| 86 |
+
INFO:root:Average loss=0.06694942450103088
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3120 | LCC = 0.7936 | SRCC = 0.7910 ]
|
| 89 |
+
INFO:root:Average loss=0.061848485173705295
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2987 | LCC = 0.8031 | SRCC = 0.8002 ]
|
| 92 |
+
INFO:root:Average loss=0.05977314920761646
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3274 | LCC = 0.7991 | SRCC = 0.7974 ]
|
| 95 |
+
INFO:root:Average loss=0.05841148921694511
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3039 | LCC = 0.8013 | SRCC = 0.7976 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2870 | LCC = 0.8221 | SRCC = 0.8224 ]
|
results_bvcc/w2v2_large_asr_en/layer12/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer12/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:04f9c1c017313ab3328b7a6db815c9180fbf9bd7746903a73d48991660515a71
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer12/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e50e5dd0821a67cae239020a0d55a9bd863f2b5c1b7db8d8f732ccf9dd190188
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer12/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.6102118430993496
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4486 | LCC = 0.7210 | SRCC = 0.7226 ]
|
| 8 |
+
INFO:root:Average loss=0.3598540024115489
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3737 | LCC = 0.7838 | SRCC = 0.7838 ]
|
| 11 |
+
INFO:root:Average loss=0.29358902401649034
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3682 | LCC = 0.7965 | SRCC = 0.7936 ]
|
| 14 |
+
INFO:root:Average loss=0.2608952222344203
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3139 | LCC = 0.8086 | SRCC = 0.8059 ]
|
| 17 |
+
INFO:root:Average loss=0.24057510724434486
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3437 | LCC = 0.8072 | SRCC = 0.8060 ]
|
| 20 |
+
INFO:root:Average loss=0.22202213547932795
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3303 | LCC = 0.8044 | SRCC = 0.8016 ]
|
| 23 |
+
INFO:root:Average loss=0.22276107336466128
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3215 | LCC = 0.8071 | SRCC = 0.8033 ]
|
| 26 |
+
INFO:root:Average loss=0.19395450770090789
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2991 | LCC = 0.8101 | SRCC = 0.8059 ]
|
| 29 |
+
INFO:root:Average loss=0.18139779625030664
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3013 | LCC = 0.8105 | SRCC = 0.8059 ]
|
| 32 |
+
INFO:root:Average loss=0.16541293788796815
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3383 | LCC = 0.8089 | SRCC = 0.8058 ]
|
| 35 |
+
INFO:root:Average loss=0.15609066302959734
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3098 | LCC = 0.8048 | SRCC = 0.8012 ]
|
| 38 |
+
INFO:root:Average loss=0.14942565073187536
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3018 | LCC = 0.8046 | SRCC = 0.8013 ]
|
| 41 |
+
INFO:root:Average loss=0.1444778493963755
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3218 | LCC = 0.8104 | SRCC = 0.8074 ]
|
| 44 |
+
INFO:root:Average loss=0.12929204392891663
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3533 | LCC = 0.8064 | SRCC = 0.8051 ]
|
| 47 |
+
INFO:root:Average loss=0.12945508059018698
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3040 | LCC = 0.8066 | SRCC = 0.8034 ]
|
| 50 |
+
INFO:root:Average loss=0.1215602186245796
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3055 | LCC = 0.8009 | SRCC = 0.7961 ]
|
| 53 |
+
INFO:root:Average loss=0.10900326655843319
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2961 | LCC = 0.8056 | SRCC = 0.8030 ]
|
| 56 |
+
INFO:root:Average loss=0.1064690011434066
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3581 | LCC = 0.8007 | SRCC = 0.8006 ]
|
| 59 |
+
INFO:root:Average loss=0.10853447392582893
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3901 | LCC = 0.8002 | SRCC = 0.8011 ]
|
| 62 |
+
INFO:root:Average loss=0.08927819065940686
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3061 | LCC = 0.8089 | SRCC = 0.8049 ]
|
| 65 |
+
INFO:root:Average loss=0.08615353049185032
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3125 | LCC = 0.8016 | SRCC = 0.7982 ]
|
| 68 |
+
INFO:root:Average loss=0.0913308357867675
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.2913 | LCC = 0.8074 | SRCC = 0.8046 ]
|
| 71 |
+
INFO:root:Average loss=0.07851233495733677
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3270 | LCC = 0.7990 | SRCC = 0.7969 ]
|
| 74 |
+
INFO:root:Average loss=0.07828337899767436
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3264 | LCC = 0.8047 | SRCC = 0.8044 ]
|
| 77 |
+
INFO:root:Average loss=0.07540841166598675
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2951 | LCC = 0.8084 | SRCC = 0.8064 ]
|
| 80 |
+
INFO:root:Average loss=0.0709887038056667
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3058 | LCC = 0.8052 | SRCC = 0.8026 ]
|
| 83 |
+
INFO:root:Average loss=0.07145582717389633
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3043 | LCC = 0.8051 | SRCC = 0.8032 ]
|
| 86 |
+
INFO:root:Average loss=0.06620600533026916
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3009 | LCC = 0.8102 | SRCC = 0.8087 ]
|
| 89 |
+
INFO:root:Average loss=0.06880018468468617
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3437 | LCC = 0.8032 | SRCC = 0.8023 ]
|
| 92 |
+
INFO:root:Average loss=0.06038604270762358
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3027 | LCC = 0.8078 | SRCC = 0.8054 ]
|
| 95 |
+
INFO:root:Average loss=0.05491556771672689
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3178 | LCC = 0.8009 | SRCC = 0.8005 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2999 | LCC = 0.8180 | SRCC = 0.8209 ]
|
results_bvcc/w2v2_large_asr_en/layer12/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer12/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0ead20be6dba04f1b5410d68166d64d5cff49fbe9a78bbfda959b188c7f6222b
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer12/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b2d9689b85d9f33cec017b4875d26aa8b002a174bb54cf0aa34e88f48d904b9
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_asr_en/layer12/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.7038338230206416
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4931 | LCC = 0.6788 | SRCC = 0.6801 ]
|
| 8 |
+
INFO:root:Average loss=0.37384560455878574
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3693 | LCC = 0.7838 | SRCC = 0.7838 ]
|
| 11 |
+
INFO:root:Average loss=0.2999933856802109
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3384 | LCC = 0.8002 | SRCC = 0.7987 ]
|
| 14 |
+
INFO:root:Average loss=0.26500188291837007
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3288 | LCC = 0.8070 | SRCC = 0.8044 ]
|
| 17 |
+
INFO:root:Average loss=0.23847780701441643
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3540 | LCC = 0.8071 | SRCC = 0.8052 ]
|
| 20 |
+
INFO:root:Average loss=0.22249199602848443
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3009 | LCC = 0.8112 | SRCC = 0.8076 ]
|
| 23 |
+
INFO:root:Average loss=0.20372035392583945
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3045 | LCC = 0.8111 | SRCC = 0.8069 ]
|
| 26 |
+
INFO:root:Average loss=0.19250235897608292
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2994 | LCC = 0.8121 | SRCC = 0.8074 ]
|
| 29 |
+
INFO:root:Average loss=0.17996701588615394
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2776 | LCC = 0.8202 | SRCC = 0.8168 ]
|
| 32 |
+
INFO:root:Average loss=0.1668266461063654
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3254 | LCC = 0.8164 | SRCC = 0.8138 ]
|
| 35 |
+
INFO:root:Average loss=0.15905942366673395
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3296 | LCC = 0.8108 | SRCC = 0.8096 ]
|
| 38 |
+
INFO:root:Average loss=0.15147841702669096
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3526 | LCC = 0.8057 | SRCC = 0.8044 ]
|
| 41 |
+
INFO:root:Average loss=0.1517454889149238
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3294 | LCC = 0.8116 | SRCC = 0.8085 ]
|
| 44 |
+
INFO:root:Average loss=0.14102680484453836
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3441 | LCC = 0.8147 | SRCC = 0.8140 ]
|
| 47 |
+
INFO:root:Average loss=0.12603258274686643
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.2879 | LCC = 0.8125 | SRCC = 0.8083 ]
|
| 50 |
+
INFO:root:Average loss=0.11384060959785412
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3237 | LCC = 0.8094 | SRCC = 0.8075 ]
|
| 53 |
+
INFO:root:Average loss=0.10927917483525398
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3029 | LCC = 0.8081 | SRCC = 0.8050 ]
|
| 56 |
+
INFO:root:Average loss=0.10366906598210335
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3050 | LCC = 0.8042 | SRCC = 0.8025 ]
|
| 59 |
+
INFO:root:Average loss=0.09829619240302306
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3236 | LCC = 0.8103 | SRCC = 0.8062 ]
|
| 62 |
+
INFO:root:Average loss=0.09296508334003963
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3247 | LCC = 0.8074 | SRCC = 0.8044 ]
|
| 65 |
+
INFO:root:Average loss=0.08567066232745464
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3026 | LCC = 0.8062 | SRCC = 0.8014 ]
|
| 68 |
+
INFO:root:Average loss=0.08407249258687863
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3115 | LCC = 0.8053 | SRCC = 0.8025 ]
|
| 71 |
+
INFO:root:Average loss=0.07837912526268226
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2995 | LCC = 0.8018 | SRCC = 0.7982 ]
|
| 74 |
+
INFO:root:Average loss=0.08392994831769894
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3279 | LCC = 0.8054 | SRCC = 0.8005 ]
|
| 77 |
+
INFO:root:Average loss=0.07431051794152993
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2946 | LCC = 0.8070 | SRCC = 0.8038 ]
|
| 80 |
+
INFO:root:Average loss=0.06799868661432694
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3061 | LCC = 0.7993 | SRCC = 0.7971 ]
|
| 83 |
+
INFO:root:Average loss=0.0673973303861343
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3030 | LCC = 0.8015 | SRCC = 0.7973 ]
|
| 86 |
+
INFO:root:Average loss=0.06311425772041847
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3116 | LCC = 0.8084 | SRCC = 0.8064 ]
|
| 89 |
+
INFO:root:Average loss=0.06494387826667382
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3072 | LCC = 0.8091 | SRCC = 0.8060 ]
|
| 92 |
+
INFO:root:Average loss=0.059478260385684475
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3006 | LCC = 0.8058 | SRCC = 0.8045 ]
|
| 95 |
+
INFO:root:Average loss=0.059121602143232636
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2950 | LCC = 0.8070 | SRCC = 0.8054 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3062 | LCC = 0.8157 | SRCC = 0.8164 ]
|
results_bvcc/w2v2_large_asr_en/layer12/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer12/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:da8afe3d99c6ced5a81feaa639a55c8e1a9b589fbf4b0a8699e50494160966a1
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer12/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8d0807b5465cb26fcd978057b9d0de1c7cdb5d993e499ad0e61943909c16ea76
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_asr_en/layer12/run4/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.629870287500895
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4628 | LCC = 0.7109 | SRCC = 0.7140 ]
|
| 8 |
+
INFO:root:Average loss=0.3486146888671777
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3833 | LCC = 0.7779 | SRCC = 0.7778 ]
|
| 11 |
+
INFO:root:Average loss=0.30247746923795116
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3435 | LCC = 0.7981 | SRCC = 0.7969 ]
|
| 14 |
+
INFO:root:Average loss=0.2903971492480009
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3342 | LCC = 0.8005 | SRCC = 0.7989 ]
|
| 17 |
+
INFO:root:Average loss=0.2524496593918556
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3906 | LCC = 0.8058 | SRCC = 0.8060 ]
|
| 20 |
+
INFO:root:Average loss=0.22911507846453252
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3181 | LCC = 0.8064 | SRCC = 0.8041 ]
|
| 23 |
+
INFO:root:Average loss=0.21391650709586266
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3005 | LCC = 0.8122 | SRCC = 0.8103 ]
|
| 26 |
+
INFO:root:Average loss=0.1907470042888935
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3211 | LCC = 0.8106 | SRCC = 0.8086 ]
|
| 29 |
+
INFO:root:Average loss=0.182094951852774
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3109 | LCC = 0.8112 | SRCC = 0.8094 ]
|
| 32 |
+
INFO:root:Average loss=0.17697418882296637
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3565 | LCC = 0.8053 | SRCC = 0.8055 ]
|
| 35 |
+
INFO:root:Average loss=0.15253328579740646
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2970 | LCC = 0.8096 | SRCC = 0.8092 ]
|
| 38 |
+
INFO:root:Average loss=0.15185313031841546
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3611 | LCC = 0.8006 | SRCC = 0.7988 ]
|
| 41 |
+
INFO:root:Average loss=0.13487681068288973
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3007 | LCC = 0.8081 | SRCC = 0.8063 ]
|
| 44 |
+
INFO:root:Average loss=0.12277116760229453
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3128 | LCC = 0.8080 | SRCC = 0.8051 ]
|
| 47 |
+
INFO:root:Average loss=0.12296248007661258
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3159 | LCC = 0.8034 | SRCC = 0.7999 ]
|
| 50 |
+
INFO:root:Average loss=0.10632928155171566
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3184 | LCC = 0.8101 | SRCC = 0.8085 ]
|
| 53 |
+
INFO:root:Average loss=0.11399256285184468
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2917 | LCC = 0.8103 | SRCC = 0.8100 ]
|
| 56 |
+
INFO:root:Average loss=0.10046655785005826
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3252 | LCC = 0.8092 | SRCC = 0.8083 ]
|
| 59 |
+
INFO:root:Average loss=0.0938771465458931
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3440 | LCC = 0.8056 | SRCC = 0.8035 ]
|
| 62 |
+
INFO:root:Average loss=0.09173524800019386
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3307 | LCC = 0.8088 | SRCC = 0.8072 ]
|
| 65 |
+
INFO:root:Average loss=0.08001724840738834
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3070 | LCC = 0.8075 | SRCC = 0.8050 ]
|
| 68 |
+
INFO:root:Average loss=0.08038300457291114
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3170 | LCC = 0.8071 | SRCC = 0.8057 ]
|
| 71 |
+
INFO:root:Average loss=0.0751321111829617
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3115 | LCC = 0.8099 | SRCC = 0.8073 ]
|
| 74 |
+
INFO:root:Average loss=0.07465068246118534
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3101 | LCC = 0.8038 | SRCC = 0.8005 ]
|
| 77 |
+
INFO:root:Average loss=0.06820170103739469
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3353 | LCC = 0.8043 | SRCC = 0.8019 ]
|
| 80 |
+
INFO:root:Average loss=0.06471494991236772
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.2934 | LCC = 0.8094 | SRCC = 0.8074 ]
|
| 83 |
+
INFO:root:Average loss=0.0650003196623845
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2972 | LCC = 0.8107 | SRCC = 0.8071 ]
|
| 86 |
+
INFO:root:Average loss=0.06452292547776149
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3088 | LCC = 0.8027 | SRCC = 0.8000 ]
|
| 89 |
+
INFO:root:Average loss=0.06001786782573431
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3289 | LCC = 0.8055 | SRCC = 0.8035 ]
|
| 92 |
+
INFO:root:Average loss=0.05812227386885729
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3155 | LCC = 0.8026 | SRCC = 0.8003 ]
|
| 95 |
+
INFO:root:Average loss=0.055573841031545244
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3125 | LCC = 0.8098 | SRCC = 0.8081 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3076 | LCC = 0.8107 | SRCC = 0.8140 ]
|
results_bvcc/w2v2_large_asr_en/layer14/run0/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer14/run0/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c7bdb91f00f6807e0aa3c8fb169b2e8a9043ac5a04b0d19b446bd071909cd2f4
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer14/run0/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3568c4c623091099662ba2f8dbdaf778a4352b39059e1c3faf36bcfa6578647b
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer14/run0/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.639124198219715
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.5824 | LCC = 0.6603 | SRCC = 0.6654 ]
|
| 8 |
+
INFO:root:Average loss=0.40829688004958326
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3725 | LCC = 0.7650 | SRCC = 0.7667 ]
|
| 11 |
+
INFO:root:Average loss=0.3168532964892877
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3455 | LCC = 0.7900 | SRCC = 0.7882 ]
|
| 14 |
+
INFO:root:Average loss=0.2736568292364096
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3320 | LCC = 0.8046 | SRCC = 0.8030 ]
|
| 17 |
+
INFO:root:Average loss=0.2780578547181227
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.4968 | LCC = 0.7974 | SRCC = 0.8015 ]
|
| 20 |
+
INFO:root:Average loss=0.24588502771579301
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3295 | LCC = 0.8075 | SRCC = 0.8025 ]
|
| 23 |
+
INFO:root:Average loss=0.2572734610010416
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3514 | LCC = 0.8064 | SRCC = 0.8056 ]
|
| 26 |
+
INFO:root:Average loss=0.21617332186836463
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3010 | LCC = 0.8103 | SRCC = 0.8057 ]
|
| 29 |
+
INFO:root:Average loss=0.19947211979291377
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3068 | LCC = 0.8131 | SRCC = 0.8103 ]
|
| 32 |
+
INFO:root:Average loss=0.18418184152016273
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3275 | LCC = 0.8129 | SRCC = 0.8107 ]
|
| 35 |
+
INFO:root:Average loss=0.18136565587841547
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3026 | LCC = 0.8115 | SRCC = 0.8093 ]
|
| 38 |
+
INFO:root:Average loss=0.17634256021716657
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3040 | LCC = 0.8105 | SRCC = 0.8073 ]
|
| 41 |
+
INFO:root:Average loss=0.16672048908777726
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.2918 | LCC = 0.8074 | SRCC = 0.8056 ]
|
| 44 |
+
INFO:root:Average loss=0.1531867396373015
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2858 | LCC = 0.8110 | SRCC = 0.8089 ]
|
| 47 |
+
INFO:root:Average loss=0.13818359909913477
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3116 | LCC = 0.8099 | SRCC = 0.8066 ]
|
| 50 |
+
INFO:root:Average loss=0.13977856437365213
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3141 | LCC = 0.8075 | SRCC = 0.8063 ]
|
| 53 |
+
INFO:root:Average loss=0.12866498329318488
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3146 | LCC = 0.8058 | SRCC = 0.8037 ]
|
| 56 |
+
INFO:root:Average loss=0.12641079996067744
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3072 | LCC = 0.8084 | SRCC = 0.8061 ]
|
| 59 |
+
INFO:root:Average loss=0.11902084497687145
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3461 | LCC = 0.8102 | SRCC = 0.8089 ]
|
| 62 |
+
INFO:root:Average loss=0.10734949982128082
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.2973 | LCC = 0.8039 | SRCC = 0.8031 ]
|
| 65 |
+
INFO:root:Average loss=0.10676883839261837
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2959 | LCC = 0.8040 | SRCC = 0.8023 ]
|
| 68 |
+
INFO:root:Average loss=0.09338671265122218
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3056 | LCC = 0.8074 | SRCC = 0.8045 ]
|
| 71 |
+
INFO:root:Average loss=0.11823369424121502
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.2920 | LCC = 0.8080 | SRCC = 0.8059 ]
|
| 74 |
+
INFO:root:Average loss=0.09307295757417496
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3335 | LCC = 0.8079 | SRCC = 0.8067 ]
|
| 77 |
+
INFO:root:Average loss=0.08407396402878639
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3107 | LCC = 0.8055 | SRCC = 0.8040 ]
|
| 80 |
+
INFO:root:Average loss=0.08697844788623162
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3272 | LCC = 0.8020 | SRCC = 0.8012 ]
|
| 83 |
+
INFO:root:Average loss=0.0841066506333076
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2910 | LCC = 0.8075 | SRCC = 0.8057 ]
|
| 86 |
+
INFO:root:Average loss=0.0758334105500044
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2999 | LCC = 0.8063 | SRCC = 0.8045 ]
|
| 89 |
+
INFO:root:Average loss=0.07849707659811546
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.2987 | LCC = 0.8069 | SRCC = 0.8055 ]
|
| 92 |
+
INFO:root:Average loss=0.07672988012050971
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3049 | LCC = 0.8049 | SRCC = 0.8026 ]
|
| 95 |
+
INFO:root:Average loss=0.06985761306415765
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.2994 | LCC = 0.8019 | SRCC = 0.8003 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3380 | LCC = 0.8146 | SRCC = 0.8211 ]
|
results_bvcc/w2v2_large_asr_en/layer14/run1/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer14/run1/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0334e35d89573d14eff4c4c7292361bddda31360cdb14bf0e31f9f9060bb5063
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer14/run1/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4e1a8ba90d97d5ef442aa7226f11e56b51a1136d1243f8b231f6d2a049dfda46
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_asr_en/layer14/run1/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.5812411392346407
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4863 | LCC = 0.7401 | SRCC = 0.7450 ]
|
| 8 |
+
INFO:root:Average loss=0.3544054925441742
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3510 | LCC = 0.7938 | SRCC = 0.7942 ]
|
| 11 |
+
INFO:root:Average loss=0.31247991896592653
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3257 | LCC = 0.8021 | SRCC = 0.7997 ]
|
| 14 |
+
INFO:root:Average loss=0.2948195548393788
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3317 | LCC = 0.8104 | SRCC = 0.8059 ]
|
| 17 |
+
INFO:root:Average loss=0.2630130499601364
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3274 | LCC = 0.8069 | SRCC = 0.8040 ]
|
| 20 |
+
INFO:root:Average loss=0.24918348724261308
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3135 | LCC = 0.8088 | SRCC = 0.8055 ]
|
| 23 |
+
INFO:root:Average loss=0.2262976987239642
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.2992 | LCC = 0.8145 | SRCC = 0.8107 ]
|
| 26 |
+
INFO:root:Average loss=0.20898243402823424
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2921 | LCC = 0.8181 | SRCC = 0.8129 ]
|
| 29 |
+
INFO:root:Average loss=0.21231674899657568
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3155 | LCC = 0.8145 | SRCC = 0.8144 ]
|
| 32 |
+
INFO:root:Average loss=0.190538444962257
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2829 | LCC = 0.8193 | SRCC = 0.8154 ]
|
| 35 |
+
INFO:root:Average loss=0.18087279672423998
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.2871 | LCC = 0.8229 | SRCC = 0.8205 ]
|
| 38 |
+
INFO:root:Average loss=0.16717880161908957
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.2927 | LCC = 0.8147 | SRCC = 0.8125 ]
|
| 41 |
+
INFO:root:Average loss=0.15654730997406519
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3133 | LCC = 0.8090 | SRCC = 0.8085 ]
|
| 44 |
+
INFO:root:Average loss=0.15454688276617956
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.2922 | LCC = 0.8134 | SRCC = 0.8109 ]
|
| 47 |
+
INFO:root:Average loss=0.14769291667602
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3003 | LCC = 0.8128 | SRCC = 0.8107 ]
|
| 50 |
+
INFO:root:Average loss=0.1449632299825167
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.2969 | LCC = 0.8185 | SRCC = 0.8157 ]
|
| 53 |
+
INFO:root:Average loss=0.1303374068095134
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.2905 | LCC = 0.8096 | SRCC = 0.8073 ]
|
| 56 |
+
INFO:root:Average loss=0.1257298872447931
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3071 | LCC = 0.8121 | SRCC = 0.8089 ]
|
| 59 |
+
INFO:root:Average loss=0.11752169378674947
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.2882 | LCC = 0.8136 | SRCC = 0.8121 ]
|
| 62 |
+
INFO:root:Average loss=0.10992229252289502
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3138 | LCC = 0.8099 | SRCC = 0.8086 ]
|
| 65 |
+
INFO:root:Average loss=0.10096019081389293
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.2823 | LCC = 0.8153 | SRCC = 0.8135 ]
|
| 68 |
+
INFO:root:Average loss=0.09729361983063893
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3079 | LCC = 0.8128 | SRCC = 0.8123 ]
|
| 71 |
+
INFO:root:Average loss=0.093237780368863
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3062 | LCC = 0.8121 | SRCC = 0.8110 ]
|
| 74 |
+
INFO:root:Average loss=0.08845462774236997
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.2951 | LCC = 0.8120 | SRCC = 0.8110 ]
|
| 77 |
+
INFO:root:Average loss=0.09973976398125672
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.2915 | LCC = 0.8118 | SRCC = 0.8111 ]
|
| 80 |
+
INFO:root:Average loss=0.08304178046110348
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3001 | LCC = 0.8034 | SRCC = 0.8015 ]
|
| 83 |
+
INFO:root:Average loss=0.07921191433874461
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.2891 | LCC = 0.8084 | SRCC = 0.8081 ]
|
| 86 |
+
INFO:root:Average loss=0.08367712767078327
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3105 | LCC = 0.8090 | SRCC = 0.8105 ]
|
| 89 |
+
INFO:root:Average loss=0.0783615472416083
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3004 | LCC = 0.8082 | SRCC = 0.8082 ]
|
| 92 |
+
INFO:root:Average loss=0.08223956723052722
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.2974 | LCC = 0.8095 | SRCC = 0.8100 ]
|
| 95 |
+
INFO:root:Average loss=0.07453711927892306
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3065 | LCC = 0.8109 | SRCC = 0.8105 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.2835 | LCC = 0.8207 | SRCC = 0.8208 ]
|
results_bvcc/w2v2_large_asr_en/layer14/run2/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer14/run2/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d3dc384c09284d751fa7edb1e270768905ff6dfd7c70bcae7e9f414d6b353c69
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer14/run2/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7cd66cf3e7d05c3b94cfe9b9c72c6beaf7479f1479d85db356f58db841f9dad7
|
| 3 |
+
size 3405062
|
results_bvcc/w2v2_large_asr_en/layer14/run2/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.6524099031319985
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4754 | LCC = 0.6931 | SRCC = 0.6955 ]
|
| 8 |
+
INFO:root:Average loss=0.35758627225191164
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3836 | LCC = 0.7808 | SRCC = 0.7829 ]
|
| 11 |
+
INFO:root:Average loss=0.3075551348618972
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3402 | LCC = 0.7947 | SRCC = 0.7927 ]
|
| 14 |
+
INFO:root:Average loss=0.2832821517800673
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3283 | LCC = 0.8025 | SRCC = 0.8005 ]
|
| 17 |
+
INFO:root:Average loss=0.2530775230664473
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3363 | LCC = 0.8030 | SRCC = 0.8020 ]
|
| 20 |
+
INFO:root:Average loss=0.24408865108703956
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3090 | LCC = 0.8028 | SRCC = 0.8009 ]
|
| 23 |
+
INFO:root:Average loss=0.2182787757080335
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3090 | LCC = 0.8111 | SRCC = 0.8079 ]
|
| 26 |
+
INFO:root:Average loss=0.2112214899598024
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.2955 | LCC = 0.8129 | SRCC = 0.8092 ]
|
| 29 |
+
INFO:root:Average loss=0.19962961322222
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.2955 | LCC = 0.8131 | SRCC = 0.8103 ]
|
| 32 |
+
INFO:root:Average loss=0.1844892527621526
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.3150 | LCC = 0.8071 | SRCC = 0.8049 ]
|
| 35 |
+
INFO:root:Average loss=0.18336004553697047
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3011 | LCC = 0.8135 | SRCC = 0.8102 ]
|
| 38 |
+
INFO:root:Average loss=0.18344622048047873
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3021 | LCC = 0.8163 | SRCC = 0.8123 ]
|
| 41 |
+
INFO:root:Average loss=0.163018013135745
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3010 | LCC = 0.8080 | SRCC = 0.8050 ]
|
| 44 |
+
INFO:root:Average loss=0.14758349515688726
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3488 | LCC = 0.7970 | SRCC = 0.7955 ]
|
| 47 |
+
INFO:root:Average loss=0.13971616642979476
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3724 | LCC = 0.8026 | SRCC = 0.8016 ]
|
| 50 |
+
INFO:root:Average loss=0.14740928291128233
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3598 | LCC = 0.8041 | SRCC = 0.8036 ]
|
| 53 |
+
INFO:root:Average loss=0.1275135239538474
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3341 | LCC = 0.8032 | SRCC = 0.8028 ]
|
| 56 |
+
INFO:root:Average loss=0.12222580993786836
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3528 | LCC = 0.7907 | SRCC = 0.7888 ]
|
| 59 |
+
INFO:root:Average loss=0.115394481863731
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3066 | LCC = 0.7961 | SRCC = 0.7935 ]
|
| 62 |
+
INFO:root:Average loss=0.11262325063729897
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3114 | LCC = 0.7992 | SRCC = 0.7955 ]
|
| 65 |
+
INFO:root:Average loss=0.10322136613420951
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3046 | LCC = 0.7978 | SRCC = 0.7954 ]
|
| 68 |
+
INFO:root:Average loss=0.10801306600945118
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3106 | LCC = 0.7995 | SRCC = 0.7975 ]
|
| 71 |
+
INFO:root:Average loss=0.09470349884568116
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3348 | LCC = 0.8007 | SRCC = 0.8006 ]
|
| 74 |
+
INFO:root:Average loss=0.09117928639245339
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3315 | LCC = 0.7981 | SRCC = 0.7962 ]
|
| 77 |
+
INFO:root:Average loss=0.0903872536161007
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3039 | LCC = 0.8068 | SRCC = 0.8055 ]
|
| 80 |
+
INFO:root:Average loss=0.08537686267533363
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3091 | LCC = 0.8008 | SRCC = 0.7980 ]
|
| 83 |
+
INFO:root:Average loss=0.08537523032954106
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3165 | LCC = 0.8010 | SRCC = 0.7998 ]
|
| 86 |
+
INFO:root:Average loss=0.08757533223774189
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.2990 | LCC = 0.8046 | SRCC = 0.8021 ]
|
| 89 |
+
INFO:root:Average loss=0.07680801588755387
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3423 | LCC = 0.8014 | SRCC = 0.8019 ]
|
| 92 |
+
INFO:root:Average loss=0.07683007342693134
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3016 | LCC = 0.8037 | SRCC = 0.8019 ]
|
| 95 |
+
INFO:root:Average loss=0.07444462165809594
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3368 | LCC = 0.8025 | SRCC = 0.8021 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3370 | LCC = 0.7984 | SRCC = 0.8052 ]
|
results_bvcc/w2v2_large_asr_en/layer14/run3/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer14/run3/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:069389fc67653cde3cc867f7aaba683212efec873d300ef5aee50aa5b581a980
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer14/run3/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ef4b777e9e71d9ac8929c8805232b3010a34334d3386463b4f3820a6efc2dfd
|
| 3 |
+
size 3404998
|
results_bvcc/w2v2_large_asr_en/layer14/run3/train.log
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
INFO:root:Num train speech clips: 4974
|
| 2 |
+
INFO:root:Num val speech clips: 1066
|
| 3 |
+
INFO:root:Num test speech clips: 1066
|
| 4 |
+
INFO:root:Device=cuda
|
| 5 |
+
INFO:root:Average loss=0.6228337616492541
|
| 6 |
+
INFO:root:
|
| 7 |
+
[Valid][0][UTT][ MSE = 0.4639 | LCC = 0.7118 | SRCC = 0.7112 ]
|
| 8 |
+
INFO:root:Average loss=0.36973212048029286
|
| 9 |
+
INFO:root:
|
| 10 |
+
[Valid][1][UTT][ MSE = 0.3787 | LCC = 0.7793 | SRCC = 0.7781 ]
|
| 11 |
+
INFO:root:Average loss=0.31610557245902526
|
| 12 |
+
INFO:root:
|
| 13 |
+
[Valid][2][UTT][ MSE = 0.3625 | LCC = 0.7884 | SRCC = 0.7869 ]
|
| 14 |
+
INFO:root:Average loss=0.2852938834291238
|
| 15 |
+
INFO:root:
|
| 16 |
+
[Valid][3][UTT][ MSE = 0.3119 | LCC = 0.8009 | SRCC = 0.7975 ]
|
| 17 |
+
INFO:root:Average loss=0.2740824589362511
|
| 18 |
+
INFO:root:
|
| 19 |
+
[Valid][4][UTT][ MSE = 0.3700 | LCC = 0.7982 | SRCC = 0.7940 ]
|
| 20 |
+
INFO:root:Average loss=0.24433110864498678
|
| 21 |
+
INFO:root:
|
| 22 |
+
[Valid][5][UTT][ MSE = 0.3307 | LCC = 0.8002 | SRCC = 0.7949 ]
|
| 23 |
+
INFO:root:Average loss=0.23941120295188364
|
| 24 |
+
INFO:root:
|
| 25 |
+
[Valid][6][UTT][ MSE = 0.3721 | LCC = 0.8017 | SRCC = 0.7983 ]
|
| 26 |
+
INFO:root:Average loss=0.2147610577253195
|
| 27 |
+
INFO:root:
|
| 28 |
+
[Valid][7][UTT][ MSE = 0.3154 | LCC = 0.8049 | SRCC = 0.8007 ]
|
| 29 |
+
INFO:root:Average loss=0.1963093800422473
|
| 30 |
+
INFO:root:
|
| 31 |
+
[Valid][8][UTT][ MSE = 0.3047 | LCC = 0.8082 | SRCC = 0.8045 ]
|
| 32 |
+
INFO:root:Average loss=0.19134491806228957
|
| 33 |
+
INFO:root:
|
| 34 |
+
[Valid][9][UTT][ MSE = 0.2998 | LCC = 0.8076 | SRCC = 0.8039 ]
|
| 35 |
+
INFO:root:Average loss=0.17910867795730248
|
| 36 |
+
INFO:root:
|
| 37 |
+
[Valid][10][UTT][ MSE = 0.3235 | LCC = 0.8018 | SRCC = 0.8003 ]
|
| 38 |
+
INFO:root:Average loss=0.1719729864062407
|
| 39 |
+
INFO:root:
|
| 40 |
+
[Valid][11][UTT][ MSE = 0.3029 | LCC = 0.8019 | SRCC = 0.7983 ]
|
| 41 |
+
INFO:root:Average loss=0.15981111025963074
|
| 42 |
+
INFO:root:
|
| 43 |
+
[Valid][12][UTT][ MSE = 0.3317 | LCC = 0.8037 | SRCC = 0.8021 ]
|
| 44 |
+
INFO:root:Average loss=0.1607825325276607
|
| 45 |
+
INFO:root:
|
| 46 |
+
[Valid][13][UTT][ MSE = 0.3118 | LCC = 0.8021 | SRCC = 0.8010 ]
|
| 47 |
+
INFO:root:Average loss=0.15734562745843178
|
| 48 |
+
INFO:root:
|
| 49 |
+
[Valid][14][UTT][ MSE = 0.3096 | LCC = 0.7990 | SRCC = 0.7966 ]
|
| 50 |
+
INFO:root:Average loss=0.1332972825337679
|
| 51 |
+
INFO:root:
|
| 52 |
+
[Valid][15][UTT][ MSE = 0.3301 | LCC = 0.7923 | SRCC = 0.7919 ]
|
| 53 |
+
INFO:root:Average loss=0.1389224115663614
|
| 54 |
+
INFO:root:
|
| 55 |
+
[Valid][16][UTT][ MSE = 0.3122 | LCC = 0.8018 | SRCC = 0.7999 ]
|
| 56 |
+
INFO:root:Average loss=0.13425574289300504
|
| 57 |
+
INFO:root:
|
| 58 |
+
[Valid][17][UTT][ MSE = 0.3329 | LCC = 0.8014 | SRCC = 0.7979 ]
|
| 59 |
+
INFO:root:Average loss=0.11482151206105183
|
| 60 |
+
INFO:root:
|
| 61 |
+
[Valid][18][UTT][ MSE = 0.3155 | LCC = 0.7997 | SRCC = 0.7972 ]
|
| 62 |
+
INFO:root:Average loss=0.11373232744443111
|
| 63 |
+
INFO:root:
|
| 64 |
+
[Valid][19][UTT][ MSE = 0.3101 | LCC = 0.8007 | SRCC = 0.7981 ]
|
| 65 |
+
INFO:root:Average loss=0.10444456740067555
|
| 66 |
+
INFO:root:
|
| 67 |
+
[Valid][20][UTT][ MSE = 0.3060 | LCC = 0.7958 | SRCC = 0.7934 ]
|
| 68 |
+
INFO:root:Average loss=0.09774937857993138
|
| 69 |
+
INFO:root:
|
| 70 |
+
[Valid][21][UTT][ MSE = 0.3169 | LCC = 0.7925 | SRCC = 0.7909 ]
|
| 71 |
+
INFO:root:Average loss=0.09432413472005954
|
| 72 |
+
INFO:root:
|
| 73 |
+
[Valid][22][UTT][ MSE = 0.3318 | LCC = 0.7895 | SRCC = 0.7878 ]
|
| 74 |
+
INFO:root:Average loss=0.09473877371503757
|
| 75 |
+
INFO:root:
|
| 76 |
+
[Valid][23][UTT][ MSE = 0.3360 | LCC = 0.7912 | SRCC = 0.7904 ]
|
| 77 |
+
INFO:root:Average loss=0.09524659310969022
|
| 78 |
+
INFO:root:
|
| 79 |
+
[Valid][24][UTT][ MSE = 0.3147 | LCC = 0.7952 | SRCC = 0.7928 ]
|
| 80 |
+
INFO:root:Average loss=0.08925897604188858
|
| 81 |
+
INFO:root:
|
| 82 |
+
[Valid][25][UTT][ MSE = 0.3078 | LCC = 0.7954 | SRCC = 0.7935 ]
|
| 83 |
+
INFO:root:Average loss=0.08056991270337349
|
| 84 |
+
INFO:root:
|
| 85 |
+
[Valid][26][UTT][ MSE = 0.3018 | LCC = 0.7978 | SRCC = 0.7968 ]
|
| 86 |
+
INFO:root:Average loss=0.07634544687775466
|
| 87 |
+
INFO:root:
|
| 88 |
+
[Valid][27][UTT][ MSE = 0.3187 | LCC = 0.7910 | SRCC = 0.7881 ]
|
| 89 |
+
INFO:root:Average loss=0.07849306880663602
|
| 90 |
+
INFO:root:
|
| 91 |
+
[Valid][28][UTT][ MSE = 0.3045 | LCC = 0.7995 | SRCC = 0.7976 ]
|
| 92 |
+
INFO:root:Average loss=0.07884880671134362
|
| 93 |
+
INFO:root:
|
| 94 |
+
[Valid][29][UTT][ MSE = 0.3241 | LCC = 0.7931 | SRCC = 0.7911 ]
|
| 95 |
+
INFO:root:Average loss=0.0762025519536856
|
| 96 |
+
INFO:root:
|
| 97 |
+
[Valid][30][UTT][ MSE = 0.3172 | LCC = 0.7931 | SRCC = 0.7910 ]
|
| 98 |
+
INFO:root:
|
| 99 |
+
[Test][31][UTT][ MSE = 0.3233 | LCC = 0.8182 | SRCC = 0.8232 ]
|
results_bvcc/w2v2_large_asr_en/layer14/run4/config.gin
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Training config on BVCC dataset.
|
| 2 |
+
|
| 3 |
+
# ------------------------------
|
| 4 |
+
# Architecture
|
| 5 |
+
# ------------------------------
|
| 6 |
+
ProjectionHead:
|
| 7 |
+
conv_channels = (32, 32)
|
| 8 |
+
dense_neurons = (64, 32, 1)
|
| 9 |
+
use_poolings = (True, True)
|
| 10 |
+
kernel_size = 5
|
| 11 |
+
pool_size = 5
|
| 12 |
+
|
| 13 |
+
# ------------------------------
|
| 14 |
+
# Dataset
|
| 15 |
+
# ------------------------------
|
| 16 |
+
Bvcc:
|
| 17 |
+
debug = False
|
| 18 |
+
|
| 19 |
+
get_dataloader:
|
| 20 |
+
num_workers = 12
|
| 21 |
+
shuffle = True
|
| 22 |
+
|
| 23 |
+
# ------------------------------
|
| 24 |
+
# Training loop
|
| 25 |
+
# ------------------------------
|
| 26 |
+
TrainingLoop:
|
| 27 |
+
num_epochs = 30
|
| 28 |
+
learning_rate = 1e-4
|
| 29 |
+
dataset_cls = @Bvcc
|
| 30 |
+
weight_decay = 0.0
|
| 31 |
+
batch_size_train = 64
|
results_bvcc/w2v2_large_asr_en/layer14/run4/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:85672245a0b51022641a0d909fb1c8b224bc0e681ed39290f994f2275b15daf4
|
| 3 |
+
size 3403339
|
results_bvcc/w2v2_large_asr_en/layer14/run4/model_best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a5afadd72191475a9b936a4ff7e854fbfa08c81774a315128368698ed55c638b
|
| 3 |
+
size 3405062
|