HopeLiang commited on
Commit
e386ce0
·
verified ·
1 Parent(s): ea7dbeb

Upload SSL Layer MOS checkpoints (part 6)

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run2/model.pt +3 -0
  2. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run2/model_best.pt +3 -0
  3. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run2/train.log +99 -0
  4. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/config.gin +31 -0
  5. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/model.pt +3 -0
  6. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/model_best.pt +3 -0
  7. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/train.log +99 -0
  8. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/config.gin +31 -0
  9. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/model.pt +3 -0
  10. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/model_best.pt +3 -0
  11. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/train.log +99 -0
  12. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/config.gin +31 -0
  13. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/model.pt +3 -0
  14. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/model_best.pt +3 -0
  15. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/train.log +99 -0
  16. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/config.gin +31 -0
  17. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/model.pt +3 -0
  18. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/model_best.pt +3 -0
  19. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/train.log +99 -0
  20. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/config.gin +31 -0
  21. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/model.pt +3 -0
  22. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/model_best.pt +3 -0
  23. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/train.log +99 -0
  24. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/config.gin +31 -0
  25. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/model.pt +3 -0
  26. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/model_best.pt +3 -0
  27. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/train.log +99 -0
  28. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/config.gin +31 -0
  29. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/model.pt +3 -0
  30. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/model_best.pt +3 -0
  31. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/train.log +99 -0
  32. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/config.gin +31 -0
  33. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/model.pt +3 -0
  34. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/model_best.pt +3 -0
  35. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/train.log +99 -0
  36. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/config.gin +31 -0
  37. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/model.pt +3 -0
  38. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/model_best.pt +3 -0
  39. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/train.log +99 -0
  40. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/config.gin +31 -0
  41. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/model.pt +3 -0
  42. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/model_best.pt +3 -0
  43. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/train.log +99 -0
  44. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/config.gin +31 -0
  45. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/model.pt +3 -0
  46. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/model_best.pt +3 -0
  47. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/train.log +99 -0
  48. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run4/config.gin +31 -0
  49. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run4/model.pt +3 -0
  50. results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run4/model_best.pt +3 -0
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d60cd3794de1d66baa13614d24fe087acd81193f9575d6dee0eafb8cc86fee95
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c148ec4598b2a0ca2c6a483bd0ed2850d30262cbfdc00d8802e714bbdbbe07a7
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/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.48212505896122027
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3767 | LCC = 0.8105 | SRCC = 0.8180 ]
8
+ INFO:root:Average loss=0.24159042499004266
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3070 | LCC = 0.8531 | SRCC = 0.8530 ]
11
+ INFO:root:Average loss=0.21392513028322122
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2711 | LCC = 0.8600 | SRCC = 0.8633 ]
14
+ INFO:root:Average loss=0.19145774669372118
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2558 | LCC = 0.8664 | SRCC = 0.8677 ]
17
+ INFO:root:Average loss=0.17701337257256874
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2499 | LCC = 0.8706 | SRCC = 0.8708 ]
20
+ INFO:root:Average loss=0.1668390225714598
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2997 | LCC = 0.8722 | SRCC = 0.8724 ]
23
+ INFO:root:Average loss=0.1563700186327482
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2658 | LCC = 0.8744 | SRCC = 0.8728 ]
26
+ INFO:root:Average loss=0.14279718085741386
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2362 | LCC = 0.8737 | SRCC = 0.8735 ]
29
+ INFO:root:Average loss=0.1286015476171787
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2080 | LCC = 0.8730 | SRCC = 0.8730 ]
32
+ INFO:root:Average loss=0.1253895195057759
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2415 | LCC = 0.8714 | SRCC = 0.8709 ]
35
+ INFO:root:Average loss=0.119387052857723
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.3019 | LCC = 0.8696 | SRCC = 0.8730 ]
38
+ INFO:root:Average loss=0.10998855263758929
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2433 | LCC = 0.8676 | SRCC = 0.8677 ]
41
+ INFO:root:Average loss=0.10329175329743287
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2288 | LCC = 0.8697 | SRCC = 0.8711 ]
44
+ INFO:root:Average loss=0.0967078777268911
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2480 | LCC = 0.8678 | SRCC = 0.8681 ]
47
+ INFO:root:Average loss=0.0947616626150333
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.3041 | LCC = 0.8686 | SRCC = 0.8702 ]
50
+ INFO:root:Average loss=0.08734149084641384
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2650 | LCC = 0.8690 | SRCC = 0.8692 ]
53
+ INFO:root:Average loss=0.08210476965476306
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2705 | LCC = 0.8658 | SRCC = 0.8680 ]
56
+ INFO:root:Average loss=0.0858790724991988
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2605 | LCC = 0.8669 | SRCC = 0.8683 ]
59
+ INFO:root:Average loss=0.08271859643551019
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2400 | LCC = 0.8669 | SRCC = 0.8677 ]
62
+ INFO:root:Average loss=0.07146724886619128
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2449 | LCC = 0.8641 | SRCC = 0.8661 ]
65
+ INFO:root:Average loss=0.06925657678108949
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2343 | LCC = 0.8644 | SRCC = 0.8657 ]
68
+ INFO:root:Average loss=0.06704354940507656
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2738 | LCC = 0.8642 | SRCC = 0.8652 ]
71
+ INFO:root:Average loss=0.07050568801470292
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2668 | LCC = 0.8640 | SRCC = 0.8649 ]
74
+ INFO:root:Average loss=0.06089034466407238
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2396 | LCC = 0.8632 | SRCC = 0.8643 ]
77
+ INFO:root:Average loss=0.05812540471267242
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2460 | LCC = 0.8622 | SRCC = 0.8633 ]
80
+ INFO:root:Average loss=0.05923682924073476
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2718 | LCC = 0.8634 | SRCC = 0.8639 ]
83
+ INFO:root:Average loss=0.056070149183655396
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2424 | LCC = 0.8621 | SRCC = 0.8632 ]
86
+ INFO:root:Average loss=0.052097282253014736
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2522 | LCC = 0.8628 | SRCC = 0.8645 ]
89
+ INFO:root:Average loss=0.051188175638134666
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2847 | LCC = 0.8603 | SRCC = 0.8611 ]
92
+ INFO:root:Average loss=0.05074002846884422
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2526 | LCC = 0.8610 | SRCC = 0.8626 ]
95
+ INFO:root:Average loss=0.048052534532661624
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2647 | LCC = 0.8614 | SRCC = 0.8628 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2438 | LCC = 0.8729 | SRCC = 0.8723 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3c9621772466bf7b22af2c71b392b66791293e07e591edf5c88a136fa4acff0
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c2920e2ab5cd64fb24703afe28923daef978fdd28d3f36fca9bf2eefb83c8f4
3
+ size 2749638
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/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.403517763202007
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3261 | LCC = 0.8120 | SRCC = 0.8157 ]
8
+ INFO:root:Average loss=0.24763199935356775
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2886 | LCC = 0.8439 | SRCC = 0.8436 ]
11
+ INFO:root:Average loss=0.21508449908250418
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2625 | LCC = 0.8551 | SRCC = 0.8547 ]
14
+ INFO:root:Average loss=0.20363809244754988
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2662 | LCC = 0.8604 | SRCC = 0.8593 ]
17
+ INFO:root:Average loss=0.1909643248296701
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2431 | LCC = 0.8619 | SRCC = 0.8608 ]
20
+ INFO:root:Average loss=0.16637271384780222
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2367 | LCC = 0.8639 | SRCC = 0.8628 ]
23
+ INFO:root:Average loss=0.16145546495532379
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2372 | LCC = 0.8634 | SRCC = 0.8633 ]
26
+ INFO:root:Average loss=0.14887363778857085
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2355 | LCC = 0.8636 | SRCC = 0.8640 ]
29
+ INFO:root:Average loss=0.1447817728114434
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2328 | LCC = 0.8648 | SRCC = 0.8654 ]
32
+ INFO:root:Average loss=0.1334422770410012
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2355 | LCC = 0.8622 | SRCC = 0.8635 ]
35
+ INFO:root:Average loss=0.12572158624728522
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2291 | LCC = 0.8623 | SRCC = 0.8632 ]
38
+ INFO:root:Average loss=0.12735722194879484
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2256 | LCC = 0.8630 | SRCC = 0.8648 ]
41
+ INFO:root:Average loss=0.1120024187824665
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2372 | LCC = 0.8591 | SRCC = 0.8614 ]
44
+ INFO:root:Average loss=0.10663311761350204
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2204 | LCC = 0.8637 | SRCC = 0.8666 ]
47
+ INFO:root:Average loss=0.10106851475743148
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2390 | LCC = 0.8581 | SRCC = 0.8605 ]
50
+ INFO:root:Average loss=0.0962457336867467
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2224 | LCC = 0.8619 | SRCC = 0.8650 ]
53
+ INFO:root:Average loss=0.09278945452891864
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2267 | LCC = 0.8607 | SRCC = 0.8645 ]
56
+ INFO:root:Average loss=0.08597430725319263
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2215 | LCC = 0.8616 | SRCC = 0.8640 ]
59
+ INFO:root:Average loss=0.08565764367962495
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2293 | LCC = 0.8603 | SRCC = 0.8628 ]
62
+ INFO:root:Average loss=0.07930731133390696
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2194 | LCC = 0.8614 | SRCC = 0.8628 ]
65
+ INFO:root:Average loss=0.07558726586210422
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2230 | LCC = 0.8621 | SRCC = 0.8645 ]
68
+ INFO:root:Average loss=0.07274589644601712
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2332 | LCC = 0.8597 | SRCC = 0.8616 ]
71
+ INFO:root:Average loss=0.07222348241469799
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2285 | LCC = 0.8617 | SRCC = 0.8630 ]
74
+ INFO:root:Average loss=0.06443445074061553
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2423 | LCC = 0.8543 | SRCC = 0.8568 ]
77
+ INFO:root:Average loss=0.06261538699842416
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2381 | LCC = 0.8573 | SRCC = 0.8606 ]
80
+ INFO:root:Average loss=0.061561480355568424
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2324 | LCC = 0.8597 | SRCC = 0.8617 ]
83
+ INFO:root:Average loss=0.05873171679484539
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2217 | LCC = 0.8598 | SRCC = 0.8619 ]
86
+ INFO:root:Average loss=0.059468637817563154
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2265 | LCC = 0.8576 | SRCC = 0.8604 ]
89
+ INFO:root:Average loss=0.05554105355762518
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2261 | LCC = 0.8598 | SRCC = 0.8625 ]
92
+ INFO:root:Average loss=0.053816242525592826
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2218 | LCC = 0.8609 | SRCC = 0.8635 ]
95
+ INFO:root:Average loss=0.05056486806521813
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2319 | LCC = 0.8578 | SRCC = 0.8594 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2384 | LCC = 0.8679 | SRCC = 0.8661 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86ccb06f8c74689c38a6844be54c98d2e987accabae067549783af6952b76ba1
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da524cb78c9911a150206922505ae3c790bbc870ed7c69552fe3963a52460c04
3
+ size 2749638
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer3/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.4551508388458154
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3547 | LCC = 0.8126 | SRCC = 0.8195 ]
8
+ INFO:root:Average loss=0.24563788699033934
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2633 | LCC = 0.8477 | SRCC = 0.8490 ]
11
+ INFO:root:Average loss=0.22253791624918962
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2699 | LCC = 0.8594 | SRCC = 0.8589 ]
14
+ INFO:root:Average loss=0.20261611082614997
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2980 | LCC = 0.8631 | SRCC = 0.8631 ]
17
+ INFO:root:Average loss=0.18679375287431937
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2468 | LCC = 0.8674 | SRCC = 0.8662 ]
20
+ INFO:root:Average loss=0.1766640214392772
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2959 | LCC = 0.8662 | SRCC = 0.8662 ]
23
+ INFO:root:Average loss=0.16893091501715857
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2464 | LCC = 0.8692 | SRCC = 0.8684 ]
26
+ INFO:root:Average loss=0.15587023731607658
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2492 | LCC = 0.8702 | SRCC = 0.8696 ]
29
+ INFO:root:Average loss=0.14170436952740717
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2697 | LCC = 0.8688 | SRCC = 0.8693 ]
32
+ INFO:root:Average loss=0.14151361517799207
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2423 | LCC = 0.8704 | SRCC = 0.8697 ]
35
+ INFO:root:Average loss=0.1240942660623636
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2586 | LCC = 0.8640 | SRCC = 0.8648 ]
38
+ INFO:root:Average loss=0.12205112629976028
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2629 | LCC = 0.8679 | SRCC = 0.8679 ]
41
+ INFO:root:Average loss=0.11233795577516922
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2502 | LCC = 0.8637 | SRCC = 0.8640 ]
44
+ INFO:root:Average loss=0.10628868582156989
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2220 | LCC = 0.8651 | SRCC = 0.8657 ]
47
+ INFO:root:Average loss=0.09721051949339035
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2459 | LCC = 0.8624 | SRCC = 0.8628 ]
50
+ INFO:root:Average loss=0.09455199577869514
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2625 | LCC = 0.8611 | SRCC = 0.8622 ]
53
+ INFO:root:Average loss=0.08903195842718467
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2253 | LCC = 0.8645 | SRCC = 0.8648 ]
56
+ INFO:root:Average loss=0.08365719287823407
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2281 | LCC = 0.8665 | SRCC = 0.8670 ]
59
+ INFO:root:Average loss=0.08212937830159298
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2630 | LCC = 0.8632 | SRCC = 0.8663 ]
62
+ INFO:root:Average loss=0.08240758040203498
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2274 | LCC = 0.8636 | SRCC = 0.8639 ]
65
+ INFO:root:Average loss=0.07459291448004735
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2406 | LCC = 0.8621 | SRCC = 0.8625 ]
68
+ INFO:root:Average loss=0.07192672082246879
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2720 | LCC = 0.8624 | SRCC = 0.8635 ]
71
+ INFO:root:Average loss=0.07326538469164799
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2480 | LCC = 0.8634 | SRCC = 0.8633 ]
74
+ INFO:root:Average loss=0.062272916141992964
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2515 | LCC = 0.8616 | SRCC = 0.8633 ]
77
+ INFO:root:Average loss=0.06356821720225689
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2588 | LCC = 0.8613 | SRCC = 0.8628 ]
80
+ INFO:root:Average loss=0.06053972760072121
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2596 | LCC = 0.8613 | SRCC = 0.8624 ]
83
+ INFO:root:Average loss=0.05774620189689673
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2594 | LCC = 0.8618 | SRCC = 0.8622 ]
86
+ INFO:root:Average loss=0.06022118270779268
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2782 | LCC = 0.8547 | SRCC = 0.8564 ]
89
+ INFO:root:Average loss=0.055904310244398236
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2515 | LCC = 0.8604 | SRCC = 0.8610 ]
92
+ INFO:root:Average loss=0.052592406049370766
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2473 | LCC = 0.8630 | SRCC = 0.8645 ]
95
+ INFO:root:Average loss=0.051705676632431835
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2558 | LCC = 0.8633 | SRCC = 0.8650 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2411 | LCC = 0.8709 | SRCC = 0.8711 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebcc46cd859d8ac4d69cea6d604adbabe0cbebb9714c8c4d5d0cce6fd05a25c5
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run0/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:607e786e0f20d96097642cab6ea07da63324a16113243dbb66953965875a7c64
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/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.3572192581800314
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.2990 | LCC = 0.8250 | SRCC = 0.8294 ]
8
+ INFO:root:Average loss=0.2406188360391519
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2557 | LCC = 0.8495 | SRCC = 0.8508 ]
11
+ INFO:root:Average loss=0.21774951502298698
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2520 | LCC = 0.8550 | SRCC = 0.8555 ]
14
+ INFO:root:Average loss=0.1942141281488614
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2529 | LCC = 0.8609 | SRCC = 0.8614 ]
17
+ INFO:root:Average loss=0.18217804454840147
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2318 | LCC = 0.8649 | SRCC = 0.8642 ]
20
+ INFO:root:Average loss=0.16384037488546127
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2206 | LCC = 0.8636 | SRCC = 0.8641 ]
23
+ INFO:root:Average loss=0.15677847331150985
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2326 | LCC = 0.8631 | SRCC = 0.8614 ]
26
+ INFO:root:Average loss=0.14156064238303748
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2275 | LCC = 0.8631 | SRCC = 0.8633 ]
29
+ INFO:root:Average loss=0.12919007184413764
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2446 | LCC = 0.8625 | SRCC = 0.8640 ]
32
+ INFO:root:Average loss=0.12635595399217728
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2200 | LCC = 0.8653 | SRCC = 0.8662 ]
35
+ INFO:root:Average loss=0.11152979253958432
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2488 | LCC = 0.8591 | SRCC = 0.8591 ]
38
+ INFO:root:Average loss=0.10957995362770863
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2467 | LCC = 0.8597 | SRCC = 0.8608 ]
41
+ INFO:root:Average loss=0.10441889871771519
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2639 | LCC = 0.8570 | SRCC = 0.8600 ]
44
+ INFO:root:Average loss=0.09284338278648181
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2341 | LCC = 0.8575 | SRCC = 0.8579 ]
47
+ INFO:root:Average loss=0.09441712059271641
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2377 | LCC = 0.8574 | SRCC = 0.8595 ]
50
+ INFO:root:Average loss=0.08448655019776943
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2315 | LCC = 0.8576 | SRCC = 0.8574 ]
53
+ INFO:root:Average loss=0.08097663440574439
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2352 | LCC = 0.8553 | SRCC = 0.8567 ]
56
+ INFO:root:Average loss=0.07481831455459961
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2270 | LCC = 0.8577 | SRCC = 0.8584 ]
59
+ INFO:root:Average loss=0.07260157569096638
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2393 | LCC = 0.8541 | SRCC = 0.8559 ]
62
+ INFO:root:Average loss=0.06735953172812095
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2560 | LCC = 0.8505 | SRCC = 0.8516 ]
65
+ INFO:root:Average loss=0.06430891509621571
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2430 | LCC = 0.8521 | SRCC = 0.8521 ]
68
+ INFO:root:Average loss=0.06036842332627529
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2381 | LCC = 0.8530 | SRCC = 0.8530 ]
71
+ INFO:root:Average loss=0.057029750174245775
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2371 | LCC = 0.8524 | SRCC = 0.8524 ]
74
+ INFO:root:Average loss=0.054139966336198345
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2436 | LCC = 0.8494 | SRCC = 0.8511 ]
77
+ INFO:root:Average loss=0.058067966085404925
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2339 | LCC = 0.8524 | SRCC = 0.8533 ]
80
+ INFO:root:Average loss=0.05401303437657845
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2404 | LCC = 0.8510 | SRCC = 0.8514 ]
83
+ INFO:root:Average loss=0.05202999477012035
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2356 | LCC = 0.8527 | SRCC = 0.8538 ]
86
+ INFO:root:Average loss=0.050806347806102194
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2448 | LCC = 0.8505 | SRCC = 0.8508 ]
89
+ INFO:root:Average loss=0.05302146397148951
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2341 | LCC = 0.8533 | SRCC = 0.8541 ]
92
+ INFO:root:Average loss=0.04574633611795994
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2468 | LCC = 0.8508 | SRCC = 0.8516 ]
95
+ INFO:root:Average loss=0.046199703469681434
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2353 | LCC = 0.8528 | SRCC = 0.8535 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2419 | LCC = 0.8587 | SRCC = 0.8598 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e00ef37f9cb1ad312ca4939acfc88b337d981cc2b99eec9bec092805be5ebf96
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run1/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da7c506c355900a70621f47dca5a770f72fb8c665061a28c39d1f2eca0048187
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/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.40165887410059953
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3370 | LCC = 0.8110 | SRCC = 0.8176 ]
8
+ INFO:root:Average loss=0.2551034534206757
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2864 | LCC = 0.8380 | SRCC = 0.8388 ]
11
+ INFO:root:Average loss=0.2264320825537046
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2783 | LCC = 0.8516 | SRCC = 0.8510 ]
14
+ INFO:root:Average loss=0.2056539114087056
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2616 | LCC = 0.8583 | SRCC = 0.8561 ]
17
+ INFO:root:Average loss=0.1841818304398121
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2413 | LCC = 0.8612 | SRCC = 0.8594 ]
20
+ INFO:root:Average loss=0.17127456850348374
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2516 | LCC = 0.8613 | SRCC = 0.8588 ]
23
+ INFO:root:Average loss=0.16312490986325803
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2361 | LCC = 0.8636 | SRCC = 0.8625 ]
26
+ INFO:root:Average loss=0.14493517319743449
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2413 | LCC = 0.8646 | SRCC = 0.8640 ]
29
+ INFO:root:Average loss=0.13562019255298835
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2309 | LCC = 0.8662 | SRCC = 0.8655 ]
32
+ INFO:root:Average loss=0.1305906085822827
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2282 | LCC = 0.8639 | SRCC = 0.8639 ]
35
+ INFO:root:Average loss=0.1189008272993259
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2379 | LCC = 0.8627 | SRCC = 0.8616 ]
38
+ INFO:root:Average loss=0.11392348364759715
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2512 | LCC = 0.8626 | SRCC = 0.8625 ]
41
+ INFO:root:Average loss=0.10430541462623157
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2261 | LCC = 0.8638 | SRCC = 0.8648 ]
44
+ INFO:root:Average loss=0.10264057455918728
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2674 | LCC = 0.8569 | SRCC = 0.8573 ]
47
+ INFO:root:Average loss=0.09011854971639621
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2496 | LCC = 0.8565 | SRCC = 0.8568 ]
50
+ INFO:root:Average loss=0.09496126629603215
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2301 | LCC = 0.8611 | SRCC = 0.8604 ]
53
+ INFO:root:Average loss=0.08675182386277577
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2584 | LCC = 0.8585 | SRCC = 0.8577 ]
56
+ INFO:root:Average loss=0.0793410558731128
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2475 | LCC = 0.8587 | SRCC = 0.8590 ]
59
+ INFO:root:Average loss=0.07912692351218982
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.3170 | LCC = 0.8556 | SRCC = 0.8572 ]
62
+ INFO:root:Average loss=0.07360794576696861
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2426 | LCC = 0.8587 | SRCC = 0.8592 ]
65
+ INFO:root:Average loss=0.06506750078346485
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2514 | LCC = 0.8573 | SRCC = 0.8566 ]
68
+ INFO:root:Average loss=0.0636703636115178
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2360 | LCC = 0.8598 | SRCC = 0.8604 ]
71
+ INFO:root:Average loss=0.059694678689806886
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2521 | LCC = 0.8571 | SRCC = 0.8574 ]
74
+ INFO:root:Average loss=0.0643480230982487
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2336 | LCC = 0.8582 | SRCC = 0.8588 ]
77
+ INFO:root:Average loss=0.058846390447937526
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2525 | LCC = 0.8565 | SRCC = 0.8561 ]
80
+ INFO:root:Average loss=0.055393779363769755
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2305 | LCC = 0.8596 | SRCC = 0.8599 ]
83
+ INFO:root:Average loss=0.05909108141293892
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2454 | LCC = 0.8560 | SRCC = 0.8564 ]
86
+ INFO:root:Average loss=0.04971969309143531
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2265 | LCC = 0.8580 | SRCC = 0.8582 ]
89
+ INFO:root:Average loss=0.05274124906804317
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2548 | LCC = 0.8561 | SRCC = 0.8575 ]
92
+ INFO:root:Average loss=0.047600057907402515
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2492 | LCC = 0.8569 | SRCC = 0.8579 ]
95
+ INFO:root:Average loss=0.04698935822130014
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2608 | LCC = 0.8570 | SRCC = 0.8583 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2366 | LCC = 0.8565 | SRCC = 0.8588 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:deb21e92806edde8f1636efcf72a3ade7392f0b8d4a615147811b20f6452e72f
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb13be8e584ae3c957cfde558aab4bbb9a6cf2ae18319fa3bb60d14dbeccc10c
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/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.35169577025450194
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3018 | LCC = 0.8219 | SRCC = 0.8307 ]
8
+ INFO:root:Average loss=0.24696448483528236
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2837 | LCC = 0.8493 | SRCC = 0.8482 ]
11
+ INFO:root:Average loss=0.2105240334685032
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2623 | LCC = 0.8594 | SRCC = 0.8617 ]
14
+ INFO:root:Average loss=0.1943549259732931
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2346 | LCC = 0.8619 | SRCC = 0.8627 ]
17
+ INFO:root:Average loss=0.17824578867890897
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2360 | LCC = 0.8643 | SRCC = 0.8632 ]
20
+ INFO:root:Average loss=0.1619350244410527
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2364 | LCC = 0.8641 | SRCC = 0.8635 ]
23
+ INFO:root:Average loss=0.15211974810331297
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2700 | LCC = 0.8614 | SRCC = 0.8646 ]
26
+ INFO:root:Average loss=0.14754558010743216
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2246 | LCC = 0.8660 | SRCC = 0.8653 ]
29
+ INFO:root:Average loss=0.13381555705116346
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2265 | LCC = 0.8643 | SRCC = 0.8636 ]
32
+ INFO:root:Average loss=0.1330194120796827
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2336 | LCC = 0.8609 | SRCC = 0.8598 ]
35
+ INFO:root:Average loss=0.113600281950755
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2329 | LCC = 0.8603 | SRCC = 0.8596 ]
38
+ INFO:root:Average loss=0.12105331923335026
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2298 | LCC = 0.8621 | SRCC = 0.8613 ]
41
+ INFO:root:Average loss=0.10036042824578591
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2263 | LCC = 0.8601 | SRCC = 0.8584 ]
44
+ INFO:root:Average loss=0.10182072384617268
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2314 | LCC = 0.8606 | SRCC = 0.8602 ]
47
+ INFO:root:Average loss=0.09473781125285687
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2310 | LCC = 0.8576 | SRCC = 0.8592 ]
50
+ INFO:root:Average loss=0.08705820329487324
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2319 | LCC = 0.8579 | SRCC = 0.8560 ]
53
+ INFO:root:Average loss=0.08563109654455613
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2329 | LCC = 0.8568 | SRCC = 0.8567 ]
56
+ INFO:root:Average loss=0.07831280372845821
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2314 | LCC = 0.8532 | SRCC = 0.8538 ]
59
+ INFO:root:Average loss=0.07201913056465295
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2296 | LCC = 0.8538 | SRCC = 0.8533 ]
62
+ INFO:root:Average loss=0.06809305189511715
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2373 | LCC = 0.8504 | SRCC = 0.8491 ]
65
+ INFO:root:Average loss=0.06498371594800399
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2267 | LCC = 0.8545 | SRCC = 0.8543 ]
68
+ INFO:root:Average loss=0.06595812155268131
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2296 | LCC = 0.8543 | SRCC = 0.8543 ]
71
+ INFO:root:Average loss=0.06538664802717857
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2268 | LCC = 0.8576 | SRCC = 0.8578 ]
74
+ INFO:root:Average loss=0.05810415176435923
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2497 | LCC = 0.8502 | SRCC = 0.8496 ]
77
+ INFO:root:Average loss=0.05614544608845161
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2337 | LCC = 0.8533 | SRCC = 0.8527 ]
80
+ INFO:root:Average loss=0.05529665674727697
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2328 | LCC = 0.8550 | SRCC = 0.8551 ]
83
+ INFO:root:Average loss=0.05160186551988889
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2337 | LCC = 0.8569 | SRCC = 0.8561 ]
86
+ INFO:root:Average loss=0.05159499486669516
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2387 | LCC = 0.8532 | SRCC = 0.8522 ]
89
+ INFO:root:Average loss=0.05016429894245588
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2413 | LCC = 0.8558 | SRCC = 0.8564 ]
92
+ INFO:root:Average loss=0.049939813713232674
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2285 | LCC = 0.8558 | SRCC = 0.8573 ]
95
+ INFO:root:Average loss=0.045125617478520445
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2283 | LCC = 0.8531 | SRCC = 0.8541 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2412 | LCC = 0.8556 | SRCC = 0.8569 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5b8f0ca6c5a820d45f4374ebb07032a93ead16d91abe9103a76d4e177274bd3
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b30eb8f2e6449acb1d952b4b864ed749e8a51bc83be2e21067ed0a5adf08dc3e
3
+ size 2749638
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/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.4263247901048416
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3341 | LCC = 0.8069 | SRCC = 0.8171 ]
8
+ INFO:root:Average loss=0.25513175225410706
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3166 | LCC = 0.8439 | SRCC = 0.8455 ]
11
+ INFO:root:Average loss=0.2196473790666996
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2450 | LCC = 0.8571 | SRCC = 0.8570 ]
14
+ INFO:root:Average loss=0.2047925625856106
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2476 | LCC = 0.8638 | SRCC = 0.8615 ]
17
+ INFO:root:Average loss=0.18182204749721748
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2253 | LCC = 0.8649 | SRCC = 0.8654 ]
20
+ INFO:root:Average loss=0.16777020138807786
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2687 | LCC = 0.8669 | SRCC = 0.8642 ]
23
+ INFO:root:Average loss=0.1529040923103308
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2244 | LCC = 0.8670 | SRCC = 0.8661 ]
26
+ INFO:root:Average loss=0.1426549872908837
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2271 | LCC = 0.8667 | SRCC = 0.8658 ]
29
+ INFO:root:Average loss=0.13602226256177977
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2409 | LCC = 0.8656 | SRCC = 0.8644 ]
32
+ INFO:root:Average loss=0.12234792648217617
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2595 | LCC = 0.8587 | SRCC = 0.8581 ]
35
+ INFO:root:Average loss=0.1162674955259531
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2263 | LCC = 0.8579 | SRCC = 0.8580 ]
38
+ INFO:root:Average loss=0.11267953671706028
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2359 | LCC = 0.8599 | SRCC = 0.8584 ]
41
+ INFO:root:Average loss=0.10379027775846995
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2269 | LCC = 0.8582 | SRCC = 0.8586 ]
44
+ INFO:root:Average loss=0.09791425835245694
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2255 | LCC = 0.8567 | SRCC = 0.8577 ]
47
+ INFO:root:Average loss=0.09592596732844146
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2295 | LCC = 0.8588 | SRCC = 0.8587 ]
50
+ INFO:root:Average loss=0.08425334879221061
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2298 | LCC = 0.8546 | SRCC = 0.8532 ]
53
+ INFO:root:Average loss=0.08025821952674633
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2456 | LCC = 0.8513 | SRCC = 0.8505 ]
56
+ INFO:root:Average loss=0.08343392835022548
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2326 | LCC = 0.8566 | SRCC = 0.8560 ]
59
+ INFO:root:Average loss=0.07468307653489785
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2361 | LCC = 0.8555 | SRCC = 0.8555 ]
62
+ INFO:root:Average loss=0.06682860679351367
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2438 | LCC = 0.8534 | SRCC = 0.8530 ]
65
+ INFO:root:Average loss=0.06633613244272195
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2390 | LCC = 0.8497 | SRCC = 0.8493 ]
68
+ INFO:root:Average loss=0.07103961238112205
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2335 | LCC = 0.8545 | SRCC = 0.8535 ]
71
+ INFO:root:Average loss=0.06813726669702774
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2272 | LCC = 0.8534 | SRCC = 0.8556 ]
74
+ INFO:root:Average loss=0.05936766902987774
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2390 | LCC = 0.8551 | SRCC = 0.8541 ]
77
+ INFO:root:Average loss=0.0550483016249461
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2504 | LCC = 0.8553 | SRCC = 0.8544 ]
80
+ INFO:root:Average loss=0.05137874006938476
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2397 | LCC = 0.8542 | SRCC = 0.8524 ]
83
+ INFO:root:Average loss=0.05169885684377872
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2353 | LCC = 0.8555 | SRCC = 0.8551 ]
86
+ INFO:root:Average loss=0.04965984797439514
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2455 | LCC = 0.8530 | SRCC = 0.8524 ]
89
+ INFO:root:Average loss=0.0517897374736957
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2654 | LCC = 0.8526 | SRCC = 0.8524 ]
92
+ INFO:root:Average loss=0.04750314476684882
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2903 | LCC = 0.8505 | SRCC = 0.8497 ]
95
+ INFO:root:Average loss=0.046954824756353326
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2455 | LCC = 0.8490 | SRCC = 0.8482 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2395 | LCC = 0.8682 | SRCC = 0.8692 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:958fa09fff657fb43010a6c579828c2d6fcda6948a6fb2726f947744c330d27c
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:056030105fc10ca00c7666e4a380d3fa911042f24e13e3cf928c460560dcb8d2
3
+ size 2749638
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer4/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.4013510105701593
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3206 | LCC = 0.8193 | SRCC = 0.8251 ]
8
+ INFO:root:Average loss=0.24679987323589814
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2881 | LCC = 0.8489 | SRCC = 0.8498 ]
11
+ INFO:root:Average loss=0.21113414603930253
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2357 | LCC = 0.8600 | SRCC = 0.8607 ]
14
+ INFO:root:Average loss=0.20985345198557928
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2500 | LCC = 0.8619 | SRCC = 0.8615 ]
17
+ INFO:root:Average loss=0.1824611308865058
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2321 | LCC = 0.8650 | SRCC = 0.8638 ]
20
+ INFO:root:Average loss=0.16225871625236976
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2584 | LCC = 0.8605 | SRCC = 0.8629 ]
23
+ INFO:root:Average loss=0.15733267214053717
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2579 | LCC = 0.8644 | SRCC = 0.8645 ]
26
+ INFO:root:Average loss=0.1450038110025418
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2435 | LCC = 0.8663 | SRCC = 0.8645 ]
29
+ INFO:root:Average loss=0.13626796666246194
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2594 | LCC = 0.8638 | SRCC = 0.8644 ]
32
+ INFO:root:Average loss=0.1275765594954674
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2345 | LCC = 0.8638 | SRCC = 0.8635 ]
35
+ INFO:root:Average loss=0.11394043381397541
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2523 | LCC = 0.8602 | SRCC = 0.8585 ]
38
+ INFO:root:Average loss=0.11694239041744134
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2534 | LCC = 0.8609 | SRCC = 0.8603 ]
41
+ INFO:root:Average loss=0.10141534148118435
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2362 | LCC = 0.8613 | SRCC = 0.8615 ]
44
+ INFO:root:Average loss=0.0958498631341335
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2309 | LCC = 0.8592 | SRCC = 0.8595 ]
47
+ INFO:root:Average loss=0.09193842934492306
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2327 | LCC = 0.8610 | SRCC = 0.8606 ]
50
+ INFO:root:Average loss=0.08798203287789455
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2558 | LCC = 0.8586 | SRCC = 0.8589 ]
53
+ INFO:root:Average loss=0.08026020176326618
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2399 | LCC = 0.8574 | SRCC = 0.8573 ]
56
+ INFO:root:Average loss=0.07814991669968152
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2344 | LCC = 0.8567 | SRCC = 0.8575 ]
59
+ INFO:root:Average loss=0.07232317987542886
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2420 | LCC = 0.8578 | SRCC = 0.8582 ]
62
+ INFO:root:Average loss=0.07560246671812657
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2247 | LCC = 0.8610 | SRCC = 0.8617 ]
65
+ INFO:root:Average loss=0.06479993142569676
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2500 | LCC = 0.8510 | SRCC = 0.8524 ]
68
+ INFO:root:Average loss=0.06703913579575527
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2480 | LCC = 0.8539 | SRCC = 0.8553 ]
71
+ INFO:root:Average loss=0.06362319403351882
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2349 | LCC = 0.8557 | SRCC = 0.8574 ]
74
+ INFO:root:Average loss=0.05556169008979431
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2415 | LCC = 0.8553 | SRCC = 0.8568 ]
77
+ INFO:root:Average loss=0.05884094994801741
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2407 | LCC = 0.8554 | SRCC = 0.8574 ]
80
+ INFO:root:Average loss=0.05600188443293938
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2349 | LCC = 0.8561 | SRCC = 0.8581 ]
83
+ INFO:root:Average loss=0.05231332129392868
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2456 | LCC = 0.8539 | SRCC = 0.8558 ]
86
+ INFO:root:Average loss=0.05194630469076145
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2452 | LCC = 0.8524 | SRCC = 0.8540 ]
89
+ INFO:root:Average loss=0.0467600287296451
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2434 | LCC = 0.8525 | SRCC = 0.8538 ]
92
+ INFO:root:Average loss=0.043897231396001116
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2432 | LCC = 0.8491 | SRCC = 0.8513 ]
95
+ INFO:root:Average loss=0.04906416966173893
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2316 | LCC = 0.8545 | SRCC = 0.8568 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2419 | LCC = 0.8703 | SRCC = 0.8689 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc09f975ea570dd8de6495633bb41e28b4c6d4f18566a8ac286eab71406c7859
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run0/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31bfb306a992a5e589efc0d6caaf910a5fca212e947669a6021791ae35b79c56
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/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.3647264561209923
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3369 | LCC = 0.8328 | SRCC = 0.8340 ]
8
+ INFO:root:Average loss=0.2450862244153634
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2823 | LCC = 0.8550 | SRCC = 0.8510 ]
11
+ INFO:root:Average loss=0.22131552088719147
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2811 | LCC = 0.8607 | SRCC = 0.8622 ]
14
+ INFO:root:Average loss=0.20576652865379286
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2710 | LCC = 0.8626 | SRCC = 0.8624 ]
17
+ INFO:root:Average loss=0.18918423899091208
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2542 | LCC = 0.8639 | SRCC = 0.8645 ]
20
+ INFO:root:Average loss=0.17924471505177328
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2549 | LCC = 0.8651 | SRCC = 0.8648 ]
23
+ INFO:root:Average loss=0.1597173732633774
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2493 | LCC = 0.8634 | SRCC = 0.8607 ]
26
+ INFO:root:Average loss=0.15701985082183129
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2403 | LCC = 0.8630 | SRCC = 0.8603 ]
29
+ INFO:root:Average loss=0.14052439079834864
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2499 | LCC = 0.8616 | SRCC = 0.8606 ]
32
+ INFO:root:Average loss=0.12632038396520492
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2446 | LCC = 0.8585 | SRCC = 0.8572 ]
35
+ INFO:root:Average loss=0.11942222905464661
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2648 | LCC = 0.8558 | SRCC = 0.8579 ]
38
+ INFO:root:Average loss=0.11359471928041714
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2774 | LCC = 0.8484 | SRCC = 0.8484 ]
41
+ INFO:root:Average loss=0.10948953042045617
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2499 | LCC = 0.8518 | SRCC = 0.8521 ]
44
+ INFO:root:Average loss=0.10058757051443443
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2358 | LCC = 0.8559 | SRCC = 0.8540 ]
47
+ INFO:root:Average loss=0.09444239783363464
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2476 | LCC = 0.8510 | SRCC = 0.8504 ]
50
+ INFO:root:Average loss=0.08680354794248557
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2495 | LCC = 0.8524 | SRCC = 0.8513 ]
53
+ INFO:root:Average loss=0.08416341503079121
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2892 | LCC = 0.8487 | SRCC = 0.8502 ]
56
+ INFO:root:Average loss=0.07809728952363515
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2460 | LCC = 0.8490 | SRCC = 0.8485 ]
59
+ INFO:root:Average loss=0.0781081745830866
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2471 | LCC = 0.8498 | SRCC = 0.8490 ]
62
+ INFO:root:Average loss=0.07456183290252319
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2315 | LCC = 0.8516 | SRCC = 0.8520 ]
65
+ INFO:root:Average loss=0.06630134319838805
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2422 | LCC = 0.8489 | SRCC = 0.8490 ]
68
+ INFO:root:Average loss=0.06708064737419288
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2452 | LCC = 0.8478 | SRCC = 0.8481 ]
71
+ INFO:root:Average loss=0.06410350593236777
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2479 | LCC = 0.8456 | SRCC = 0.8446 ]
74
+ INFO:root:Average loss=0.060496386809226796
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2477 | LCC = 0.8476 | SRCC = 0.8480 ]
77
+ INFO:root:Average loss=0.06087472643225621
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2350 | LCC = 0.8523 | SRCC = 0.8517 ]
80
+ INFO:root:Average loss=0.05840050056576729
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2441 | LCC = 0.8481 | SRCC = 0.8484 ]
83
+ INFO:root:Average loss=0.05399366649679649
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2451 | LCC = 0.8481 | SRCC = 0.8485 ]
86
+ INFO:root:Average loss=0.05104239884381875
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2514 | LCC = 0.8447 | SRCC = 0.8454 ]
89
+ INFO:root:Average loss=0.046379657390599065
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2477 | LCC = 0.8463 | SRCC = 0.8453 ]
92
+ INFO:root:Average loss=0.04524605465718569
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2438 | LCC = 0.8480 | SRCC = 0.8468 ]
95
+ INFO:root:Average loss=0.049099207640840456
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2414 | LCC = 0.8479 | SRCC = 0.8470 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2496 | LCC = 0.8601 | SRCC = 0.8602 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6edcd92e8f241b1f39f8abbe2248e38b3de0dc1ee4f8ca2bc0a5b86cc3510211
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run1/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79b99475319d9492f9f4f13909f52a48f31a58cdfeedc1a7abb06b0969bd5ebf
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/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.36341217427681655
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3052 | LCC = 0.8347 | SRCC = 0.8364 ]
8
+ INFO:root:Average loss=0.2456711721725953
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2770 | LCC = 0.8508 | SRCC = 0.8501 ]
11
+ INFO:root:Average loss=0.22876097242801618
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2806 | LCC = 0.8600 | SRCC = 0.8571 ]
14
+ INFO:root:Average loss=0.19449595438364226
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2520 | LCC = 0.8658 | SRCC = 0.8634 ]
17
+ INFO:root:Average loss=0.18043607282332885
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2403 | LCC = 0.8666 | SRCC = 0.8662 ]
20
+ INFO:root:Average loss=0.16845933796885687
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2698 | LCC = 0.8684 | SRCC = 0.8658 ]
23
+ INFO:root:Average loss=0.15109611665591216
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2343 | LCC = 0.8631 | SRCC = 0.8624 ]
26
+ INFO:root:Average loss=0.14470888979924032
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.3042 | LCC = 0.8581 | SRCC = 0.8587 ]
29
+ INFO:root:Average loss=0.13583920628596574
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2527 | LCC = 0.8610 | SRCC = 0.8597 ]
32
+ INFO:root:Average loss=0.12527909263586387
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2546 | LCC = 0.8603 | SRCC = 0.8582 ]
35
+ INFO:root:Average loss=0.11729279284675916
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2624 | LCC = 0.8578 | SRCC = 0.8570 ]
38
+ INFO:root:Average loss=0.11554487584493099
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2687 | LCC = 0.8585 | SRCC = 0.8567 ]
41
+ INFO:root:Average loss=0.1017474483889647
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2518 | LCC = 0.8550 | SRCC = 0.8552 ]
44
+ INFO:root:Average loss=0.09209378579488167
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2672 | LCC = 0.8570 | SRCC = 0.8562 ]
47
+ INFO:root:Average loss=0.0961250891097081
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2327 | LCC = 0.8548 | SRCC = 0.8546 ]
50
+ INFO:root:Average loss=0.08398169837892056
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2401 | LCC = 0.8552 | SRCC = 0.8550 ]
53
+ INFO:root:Average loss=0.08130157533555459
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2373 | LCC = 0.8544 | SRCC = 0.8550 ]
56
+ INFO:root:Average loss=0.07885749332415752
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2506 | LCC = 0.8521 | SRCC = 0.8507 ]
59
+ INFO:root:Average loss=0.0726623892879639
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2421 | LCC = 0.8506 | SRCC = 0.8506 ]
62
+ INFO:root:Average loss=0.067769757591379
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2533 | LCC = 0.8547 | SRCC = 0.8540 ]
65
+ INFO:root:Average loss=0.06732550139228503
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2397 | LCC = 0.8534 | SRCC = 0.8527 ]
68
+ INFO:root:Average loss=0.06061897364755472
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2433 | LCC = 0.8513 | SRCC = 0.8500 ]
71
+ INFO:root:Average loss=0.05890513985202862
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2651 | LCC = 0.8465 | SRCC = 0.8460 ]
74
+ INFO:root:Average loss=0.05775275702277819
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2444 | LCC = 0.8517 | SRCC = 0.8512 ]
77
+ INFO:root:Average loss=0.05580917048530701
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2351 | LCC = 0.8527 | SRCC = 0.8524 ]
80
+ INFO:root:Average loss=0.054117803437969625
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2570 | LCC = 0.8521 | SRCC = 0.8511 ]
83
+ INFO:root:Average loss=0.05272514511568424
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2490 | LCC = 0.8512 | SRCC = 0.8503 ]
86
+ INFO:root:Average loss=0.05023255418890562
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2417 | LCC = 0.8495 | SRCC = 0.8488 ]
89
+ INFO:root:Average loss=0.048341174490558796
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2457 | LCC = 0.8500 | SRCC = 0.8482 ]
92
+ INFO:root:Average loss=0.05171361782898506
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2507 | LCC = 0.8501 | SRCC = 0.8475 ]
95
+ INFO:root:Average loss=0.04439792696100015
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2393 | LCC = 0.8525 | SRCC = 0.8513 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2752 | LCC = 0.8590 | SRCC = 0.8597 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08e8b0573262bd617b63ee395cd37d97ec4b24f3636ccf53738d2f2d81ffafdf
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be62dfaba70922177033b2150da16a5e3ab228354055eaeec3db2d21060be666
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run2/train.log ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO:root:Num train speech clips: 4974
2
+ INFO:root:Num val speech clips: 1066
3
+ INFO:root:Num test speech clips: 1066
4
+ INFO:root:Device=cuda
5
+ INFO:root:Average loss=0.3910743683958665
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.2994 | LCC = 0.8279 | SRCC = 0.8289 ]
8
+ INFO:root:Average loss=0.2482904673386843
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3127 | LCC = 0.8494 | SRCC = 0.8471 ]
11
+ INFO:root:Average loss=0.22257250165328002
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2769 | LCC = 0.8563 | SRCC = 0.8557 ]
14
+ INFO:root:Average loss=0.20747650806338358
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2727 | LCC = 0.8606 | SRCC = 0.8617 ]
17
+ INFO:root:Average loss=0.1903773407714489
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2460 | LCC = 0.8674 | SRCC = 0.8654 ]
20
+ INFO:root:Average loss=0.1798965913745073
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2297 | LCC = 0.8658 | SRCC = 0.8654 ]
23
+ INFO:root:Average loss=0.16481705525746712
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2283 | LCC = 0.8657 | SRCC = 0.8646 ]
26
+ INFO:root:Average loss=0.15823762195232588
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2425 | LCC = 0.8645 | SRCC = 0.8646 ]
29
+ INFO:root:Average loss=0.14712723774405626
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2633 | LCC = 0.8622 | SRCC = 0.8617 ]
32
+ INFO:root:Average loss=0.137577229203322
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2257 | LCC = 0.8612 | SRCC = 0.8609 ]
35
+ INFO:root:Average loss=0.12150792701122089
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2346 | LCC = 0.8558 | SRCC = 0.8553 ]
38
+ INFO:root:Average loss=0.11626947489686501
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2365 | LCC = 0.8563 | SRCC = 0.8569 ]
41
+ INFO:root:Average loss=0.11079482380778362
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2468 | LCC = 0.8491 | SRCC = 0.8498 ]
44
+ INFO:root:Average loss=0.10233893474707237
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2654 | LCC = 0.8504 | SRCC = 0.8500 ]
47
+ INFO:root:Average loss=0.09710497638353935
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2451 | LCC = 0.8523 | SRCC = 0.8514 ]
50
+ INFO:root:Average loss=0.08908081900041837
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2518 | LCC = 0.8505 | SRCC = 0.8510 ]
53
+ INFO:root:Average loss=0.08982657645948422
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2517 | LCC = 0.8472 | SRCC = 0.8478 ]
56
+ INFO:root:Average loss=0.08223194442689419
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2398 | LCC = 0.8512 | SRCC = 0.8511 ]
59
+ INFO:root:Average loss=0.08082248733784908
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2622 | LCC = 0.8478 | SRCC = 0.8479 ]
62
+ INFO:root:Average loss=0.07176543657596295
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2849 | LCC = 0.8428 | SRCC = 0.8425 ]
65
+ INFO:root:Average loss=0.07014165833019294
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2666 | LCC = 0.8426 | SRCC = 0.8417 ]
68
+ INFO:root:Average loss=0.067525883563436
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2464 | LCC = 0.8464 | SRCC = 0.8450 ]
71
+ INFO:root:Average loss=0.06632072597933121
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2489 | LCC = 0.8459 | SRCC = 0.8463 ]
74
+ INFO:root:Average loss=0.07072250764721479
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2553 | LCC = 0.8481 | SRCC = 0.8466 ]
77
+ INFO:root:Average loss=0.05845999961289076
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2555 | LCC = 0.8432 | SRCC = 0.8416 ]
80
+ INFO:root:Average loss=0.05378421171544454
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2618 | LCC = 0.8441 | SRCC = 0.8429 ]
83
+ INFO:root:Average loss=0.053914408366649576
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2450 | LCC = 0.8444 | SRCC = 0.8448 ]
86
+ INFO:root:Average loss=0.05011503995420077
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2563 | LCC = 0.8434 | SRCC = 0.8427 ]
89
+ INFO:root:Average loss=0.05067756752937268
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2394 | LCC = 0.8503 | SRCC = 0.8510 ]
92
+ INFO:root:Average loss=0.05002134388838059
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2468 | LCC = 0.8464 | SRCC = 0.8468 ]
95
+ INFO:root:Average loss=0.04614516285558542
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2442 | LCC = 0.8477 | SRCC = 0.8477 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2500 | LCC = 0.8586 | SRCC = 0.8583 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:032cda5e208ec0c92aba7fd2a684d435143dc5cd174d8d263ec202233a3fb92d
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:142b530226ced77155d0639b58fa5692e277369bb6c04e512e1ffb9c8f21e3e8
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run3/train.log ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO:root:Num train speech clips: 4974
2
+ INFO:root:Num val speech clips: 1066
3
+ INFO:root:Num test speech clips: 1066
4
+ INFO:root:Device=cuda
5
+ INFO:root:Average loss=0.45257461338471144
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3301 | LCC = 0.8254 | SRCC = 0.8307 ]
8
+ INFO:root:Average loss=0.24987471543061426
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2638 | LCC = 0.8459 | SRCC = 0.8506 ]
11
+ INFO:root:Average loss=0.21798543517406171
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2613 | LCC = 0.8593 | SRCC = 0.8598 ]
14
+ INFO:root:Average loss=0.19310585953868353
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2482 | LCC = 0.8644 | SRCC = 0.8641 ]
17
+ INFO:root:Average loss=0.17893173593359116
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2462 | LCC = 0.8615 | SRCC = 0.8588 ]
20
+ INFO:root:Average loss=0.1604620206814546
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2317 | LCC = 0.8629 | SRCC = 0.8613 ]
23
+ INFO:root:Average loss=0.14410021853370544
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2373 | LCC = 0.8595 | SRCC = 0.8571 ]
26
+ INFO:root:Average loss=0.1442149166877453
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2284 | LCC = 0.8592 | SRCC = 0.8577 ]
29
+ INFO:root:Average loss=0.13009902242666635
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2336 | LCC = 0.8574 | SRCC = 0.8584 ]
32
+ INFO:root:Average loss=0.12277384408009358
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2321 | LCC = 0.8586 | SRCC = 0.8586 ]
35
+ INFO:root:Average loss=0.1101332270564177
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2406 | LCC = 0.8533 | SRCC = 0.8537 ]
38
+ INFO:root:Average loss=0.1030655565361182
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2380 | LCC = 0.8528 | SRCC = 0.8520 ]
41
+ INFO:root:Average loss=0.10142424148626816
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2334 | LCC = 0.8537 | SRCC = 0.8528 ]
44
+ INFO:root:Average loss=0.0928691405420884
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2425 | LCC = 0.8513 | SRCC = 0.8509 ]
47
+ INFO:root:Average loss=0.08392868463236552
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2499 | LCC = 0.8502 | SRCC = 0.8493 ]
50
+ INFO:root:Average loss=0.07562753782631496
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2454 | LCC = 0.8471 | SRCC = 0.8459 ]
53
+ INFO:root:Average loss=0.07394624014313404
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2429 | LCC = 0.8473 | SRCC = 0.8470 ]
56
+ INFO:root:Average loss=0.06798708797074281
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2501 | LCC = 0.8445 | SRCC = 0.8442 ]
59
+ INFO:root:Average loss=0.06676959738326378
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2391 | LCC = 0.8488 | SRCC = 0.8476 ]
62
+ INFO:root:Average loss=0.06508324992580292
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2480 | LCC = 0.8432 | SRCC = 0.8424 ]
65
+ INFO:root:Average loss=0.061816331811058216
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2513 | LCC = 0.8446 | SRCC = 0.8443 ]
68
+ INFO:root:Average loss=0.060355358160077
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2285 | LCC = 0.8512 | SRCC = 0.8527 ]
71
+ INFO:root:Average loss=0.05913594076171135
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2445 | LCC = 0.8465 | SRCC = 0.8452 ]
74
+ INFO:root:Average loss=0.05481694161127775
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2351 | LCC = 0.8480 | SRCC = 0.8471 ]
77
+ INFO:root:Average loss=0.05041407796148306
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2406 | LCC = 0.8464 | SRCC = 0.8468 ]
80
+ INFO:root:Average loss=0.05042444432202058
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2393 | LCC = 0.8465 | SRCC = 0.8461 ]
83
+ INFO:root:Average loss=0.04690109766446627
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2425 | LCC = 0.8454 | SRCC = 0.8454 ]
86
+ INFO:root:Average loss=0.04716737792851069
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2463 | LCC = 0.8431 | SRCC = 0.8430 ]
89
+ INFO:root:Average loss=0.0466637581300277
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2567 | LCC = 0.8460 | SRCC = 0.8493 ]
92
+ INFO:root:Average loss=0.041203983271351226
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2431 | LCC = 0.8455 | SRCC = 0.8452 ]
95
+ INFO:root:Average loss=0.04254784334737521
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2443 | LCC = 0.8445 | SRCC = 0.8454 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2443 | LCC = 0.8644 | SRCC = 0.8621 ]
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run4/config.gin ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training config on BVCC dataset.
2
+
3
+ # ------------------------------
4
+ # Architecture
5
+ # ------------------------------
6
+ ProjectionHead:
7
+ conv_channels = (32, 32)
8
+ dense_neurons = (64, 32, 1)
9
+ use_poolings = (True, True)
10
+ kernel_size = 5
11
+ pool_size = 5
12
+
13
+ # ------------------------------
14
+ # Dataset
15
+ # ------------------------------
16
+ Bvcc:
17
+ debug = False
18
+
19
+ get_dataloader:
20
+ num_workers = 12
21
+ shuffle = True
22
+
23
+ # ------------------------------
24
+ # Training loop
25
+ # ------------------------------
26
+ TrainingLoop:
27
+ num_epochs = 30
28
+ learning_rate = 1e-4
29
+ dataset_cls = @Bvcc
30
+ weight_decay = 0.0
31
+ batch_size_train = 64
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4a861c4876b0d32e8af1f90ddfb8c8b087c413c50c70792d90fbc02309ef78a
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_4epoch/layer5/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6e4106e0b829ed7399b5bbc4d1290e10a6806cf25cdf6e2b422952a933aa0f6
3
+ size 2749702