HopeLiang commited on
Commit
632b39f
·
verified ·
1 Parent(s): 763e9e0

Upload SSL Layer MOS checkpoints (part 10)

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_large_xlsr53/layer20/run2/model.pt +3 -0
  2. results_bvcc/w2v2_large_xlsr53/layer20/run2/model_best.pt +3 -0
  3. results_bvcc/w2v2_large_xlsr53/layer20/run2/train.log +99 -0
  4. results_bvcc/w2v2_large_xlsr53/layer20/run3/config.gin +31 -0
  5. results_bvcc/w2v2_large_xlsr53/layer20/run3/model.pt +3 -0
  6. results_bvcc/w2v2_large_xlsr53/layer20/run3/model_best.pt +3 -0
  7. results_bvcc/w2v2_large_xlsr53/layer20/run3/train.log +99 -0
  8. results_bvcc/w2v2_large_xlsr53/layer20/run4/config.gin +31 -0
  9. results_bvcc/w2v2_large_xlsr53/layer20/run4/model.pt +3 -0
  10. results_bvcc/w2v2_large_xlsr53/layer20/run4/model_best.pt +3 -0
  11. results_bvcc/w2v2_large_xlsr53/layer20/run4/train.log +99 -0
  12. results_bvcc/w2v2_large_xlsr53/layer22/run0/config.gin +31 -0
  13. results_bvcc/w2v2_large_xlsr53/layer22/run0/model.pt +3 -0
  14. results_bvcc/w2v2_large_xlsr53/layer22/run0/model_best.pt +3 -0
  15. results_bvcc/w2v2_large_xlsr53/layer22/run0/train.log +99 -0
  16. results_bvcc/w2v2_large_xlsr53/layer22/run1/config.gin +31 -0
  17. results_bvcc/w2v2_large_xlsr53/layer22/run1/model.pt +3 -0
  18. results_bvcc/w2v2_large_xlsr53/layer22/run1/model_best.pt +3 -0
  19. results_bvcc/w2v2_large_xlsr53/layer22/run1/train.log +99 -0
  20. results_bvcc/w2v2_large_xlsr53/layer22/run2/config.gin +31 -0
  21. results_bvcc/w2v2_large_xlsr53/layer22/run2/model.pt +3 -0
  22. results_bvcc/w2v2_large_xlsr53/layer22/run2/model_best.pt +3 -0
  23. results_bvcc/w2v2_large_xlsr53/layer22/run2/train.log +99 -0
  24. results_bvcc/w2v2_large_xlsr53/layer22/run3/config.gin +31 -0
  25. results_bvcc/w2v2_large_xlsr53/layer22/run3/model.pt +3 -0
  26. results_bvcc/w2v2_large_xlsr53/layer22/run3/model_best.pt +3 -0
  27. results_bvcc/w2v2_large_xlsr53/layer22/run3/train.log +99 -0
  28. results_bvcc/w2v2_large_xlsr53/layer22/run4/config.gin +31 -0
  29. results_bvcc/w2v2_large_xlsr53/layer22/run4/model.pt +3 -0
  30. results_bvcc/w2v2_large_xlsr53/layer22/run4/model_best.pt +3 -0
  31. results_bvcc/w2v2_large_xlsr53/layer22/run4/train.log +99 -0
  32. results_bvcc/w2v2_large_xlsr53/layer4/run0/config.gin +31 -0
  33. results_bvcc/w2v2_large_xlsr53/layer4/run0/model.pt +3 -0
  34. results_bvcc/w2v2_large_xlsr53/layer4/run0/model_best.pt +3 -0
  35. results_bvcc/w2v2_large_xlsr53/layer4/run0/train.log +99 -0
  36. results_bvcc/w2v2_large_xlsr53/layer4/run1/config.gin +31 -0
  37. results_bvcc/w2v2_large_xlsr53/layer4/run1/model.pt +3 -0
  38. results_bvcc/w2v2_large_xlsr53/layer4/run1/model_best.pt +3 -0
  39. results_bvcc/w2v2_large_xlsr53/layer4/run1/train.log +99 -0
  40. results_bvcc/w2v2_large_xlsr53/layer4/run2/config.gin +31 -0
  41. results_bvcc/w2v2_large_xlsr53/layer4/run2/model.pt +3 -0
  42. results_bvcc/w2v2_large_xlsr53/layer4/run2/model_best.pt +3 -0
  43. results_bvcc/w2v2_large_xlsr53/layer4/run2/train.log +99 -0
  44. results_bvcc/w2v2_large_xlsr53/layer4/run3/config.gin +31 -0
  45. results_bvcc/w2v2_large_xlsr53/layer4/run3/model.pt +3 -0
  46. results_bvcc/w2v2_large_xlsr53/layer4/run3/model_best.pt +3 -0
  47. results_bvcc/w2v2_large_xlsr53/layer4/run3/train.log +99 -0
  48. results_bvcc/w2v2_large_xlsr53/layer4/run4/config.gin +31 -0
  49. results_bvcc/w2v2_large_xlsr53/layer4/run4/model.pt +3 -0
  50. results_bvcc/w2v2_large_xlsr53/layer4/run4/model_best.pt +3 -0
results_bvcc/w2v2_large_xlsr53/layer20/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d14589be817c0d74382b4c45df57ba4f96a2d9a394b0defdfa20c7967059784
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer20/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31e283b49ce56846e5ea60201716f2775e95f09dbc524bb890f76e317df5e738
3
+ size 3404998
results_bvcc/w2v2_large_xlsr53/layer20/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.5037704167457727
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3710 | LCC = 0.8137 | SRCC = 0.8130 ]
8
+ INFO:root:Average loss=0.3218344642947882
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3017 | LCC = 0.8344 | SRCC = 0.8325 ]
11
+ INFO:root:Average loss=0.28683051887231
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3121 | LCC = 0.8354 | SRCC = 0.8332 ]
14
+ INFO:root:Average loss=0.2438610845651382
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.3209 | LCC = 0.8407 | SRCC = 0.8401 ]
17
+ INFO:root:Average loss=0.24924463224716675
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2590 | LCC = 0.8458 | SRCC = 0.8458 ]
20
+ INFO:root:Average loss=0.2369476317977294
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2576 | LCC = 0.8437 | SRCC = 0.8408 ]
23
+ INFO:root:Average loss=0.22474074516540918
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2660 | LCC = 0.8524 | SRCC = 0.8500 ]
26
+ INFO:root:Average loss=0.2444543399107762
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2875 | LCC = 0.8515 | SRCC = 0.8516 ]
29
+ INFO:root:Average loss=0.20904724892133322
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2772 | LCC = 0.8513 | SRCC = 0.8500 ]
32
+ INFO:root:Average loss=0.22854098400626427
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2458 | LCC = 0.8576 | SRCC = 0.8539 ]
35
+ INFO:root:Average loss=0.20622509364516306
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2966 | LCC = 0.8542 | SRCC = 0.8541 ]
38
+ INFO:root:Average loss=0.24844215485530022
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.4012 | LCC = 0.8432 | SRCC = 0.8474 ]
41
+ INFO:root:Average loss=0.23113030615525368
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2578 | LCC = 0.8575 | SRCC = 0.8540 ]
44
+ INFO:root:Average loss=0.20823164876454917
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2423 | LCC = 0.8570 | SRCC = 0.8573 ]
47
+ INFO:root:Average loss=0.19476968174179396
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2662 | LCC = 0.8567 | SRCC = 0.8562 ]
50
+ INFO:root:Average loss=0.20127781767111558
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2434 | LCC = 0.8608 | SRCC = 0.8588 ]
53
+ INFO:root:Average loss=0.1766820180301483
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2376 | LCC = 0.8623 | SRCC = 0.8589 ]
56
+ INFO:root:Average loss=0.17697289051153722
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2839 | LCC = 0.8550 | SRCC = 0.8540 ]
59
+ INFO:root:Average loss=0.17354973529775938
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2558 | LCC = 0.8608 | SRCC = 0.8563 ]
62
+ INFO:root:Average loss=0.17916993128183561
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2381 | LCC = 0.8615 | SRCC = 0.8606 ]
65
+ INFO:root:Average loss=0.17042449068946716
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.3470 | LCC = 0.8488 | SRCC = 0.8525 ]
68
+ INFO:root:Average loss=0.1760022763449412
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2639 | LCC = 0.8549 | SRCC = 0.8551 ]
71
+ INFO:root:Average loss=0.16853642702484742
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2303 | LCC = 0.8604 | SRCC = 0.8592 ]
74
+ INFO:root:Average loss=0.15448664902494505
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2323 | LCC = 0.8632 | SRCC = 0.8597 ]
77
+ INFO:root:Average loss=0.15746389988523263
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.3181 | LCC = 0.8536 | SRCC = 0.8569 ]
80
+ INFO:root:Average loss=0.15692691256602606
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2275 | LCC = 0.8571 | SRCC = 0.8586 ]
83
+ INFO:root:Average loss=0.14679662816417524
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2730 | LCC = 0.8571 | SRCC = 0.8559 ]
86
+ INFO:root:Average loss=0.13968717698485422
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2873 | LCC = 0.8537 | SRCC = 0.8529 ]
89
+ INFO:root:Average loss=0.14161112551123667
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2696 | LCC = 0.8556 | SRCC = 0.8553 ]
92
+ INFO:root:Average loss=0.13464587267774802
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2552 | LCC = 0.8610 | SRCC = 0.8577 ]
95
+ INFO:root:Average loss=0.134915314328212
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2346 | LCC = 0.8566 | SRCC = 0.8554 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2229 | LCC = 0.8692 | SRCC = 0.8727 ]
results_bvcc/w2v2_large_xlsr53/layer20/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_xlsr53/layer20/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7b5850ead9cca3ebc32292cd021f6c60bc35c88612ac958069def7a13b4be03
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer20/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48420bd691f9a26d2a20781ee333fe6be696d58cbe7c68542d5149ad2b63fc7c
3
+ size 3404998
results_bvcc/w2v2_large_xlsr53/layer20/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.5425823359535291
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3698 | LCC = 0.8129 | SRCC = 0.8186 ]
8
+ INFO:root:Average loss=0.32132799713275373
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3541 | LCC = 0.8196 | SRCC = 0.8255 ]
11
+ INFO:root:Average loss=0.2719362173707057
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3146 | LCC = 0.8396 | SRCC = 0.8391 ]
14
+ INFO:root:Average loss=0.2679524010954759
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.3305 | LCC = 0.8328 | SRCC = 0.8417 ]
17
+ INFO:root:Average loss=0.25979573260515165
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2647 | LCC = 0.8423 | SRCC = 0.8431 ]
20
+ INFO:root:Average loss=0.2486117742000482
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2708 | LCC = 0.8522 | SRCC = 0.8505 ]
23
+ INFO:root:Average loss=0.24040728425368285
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2817 | LCC = 0.8404 | SRCC = 0.8411 ]
26
+ INFO:root:Average loss=0.22360487377796417
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2636 | LCC = 0.8546 | SRCC = 0.8516 ]
29
+ INFO:root:Average loss=0.2194032772229268
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2495 | LCC = 0.8537 | SRCC = 0.8512 ]
32
+ INFO:root:Average loss=0.2205862891024504
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2866 | LCC = 0.8585 | SRCC = 0.8552 ]
35
+ INFO:root:Average loss=0.20756436807987017
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2407 | LCC = 0.8547 | SRCC = 0.8531 ]
38
+ INFO:root:Average loss=0.2210563491934385
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2822 | LCC = 0.8585 | SRCC = 0.8546 ]
41
+ INFO:root:Average loss=0.20477527876694998
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2332 | LCC = 0.8603 | SRCC = 0.8590 ]
44
+ INFO:root:Average loss=0.19850699737285957
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2565 | LCC = 0.8544 | SRCC = 0.8524 ]
47
+ INFO:root:Average loss=0.1866154365050487
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2960 | LCC = 0.8608 | SRCC = 0.8589 ]
50
+ INFO:root:Average loss=0.18801097056040397
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2596 | LCC = 0.8573 | SRCC = 0.8543 ]
53
+ INFO:root:Average loss=0.1865861190435214
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2432 | LCC = 0.8593 | SRCC = 0.8560 ]
56
+ INFO:root:Average loss=0.19375459047464225
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2574 | LCC = 0.8593 | SRCC = 0.8575 ]
59
+ INFO:root:Average loss=0.18797000516683626
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2607 | LCC = 0.8636 | SRCC = 0.8605 ]
62
+ INFO:root:Average loss=0.1724727220642261
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2477 | LCC = 0.8620 | SRCC = 0.8591 ]
65
+ INFO:root:Average loss=0.16185139960203415
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2719 | LCC = 0.8507 | SRCC = 0.8535 ]
68
+ INFO:root:Average loss=0.1838340662801877
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2447 | LCC = 0.8636 | SRCC = 0.8593 ]
71
+ INFO:root:Average loss=0.17008591290467825
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.3163 | LCC = 0.8507 | SRCC = 0.8530 ]
74
+ INFO:root:Average loss=0.16352493315935135
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2311 | LCC = 0.8599 | SRCC = 0.8568 ]
77
+ INFO:root:Average loss=0.15196033337941536
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2397 | LCC = 0.8593 | SRCC = 0.8537 ]
80
+ INFO:root:Average loss=0.14883078491458526
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2508 | LCC = 0.8544 | SRCC = 0.8538 ]
83
+ INFO:root:Average loss=0.15664823410602716
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2321 | LCC = 0.8595 | SRCC = 0.8555 ]
86
+ INFO:root:Average loss=0.1461077043070243
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2510 | LCC = 0.8575 | SRCC = 0.8537 ]
89
+ INFO:root:Average loss=0.13786816090727463
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2476 | LCC = 0.8555 | SRCC = 0.8539 ]
92
+ INFO:root:Average loss=0.13456128919735932
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2449 | LCC = 0.8544 | SRCC = 0.8482 ]
95
+ INFO:root:Average loss=0.13000222629843614
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2385 | LCC = 0.8561 | SRCC = 0.8531 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2623 | LCC = 0.8731 | SRCC = 0.8742 ]
results_bvcc/w2v2_large_xlsr53/layer20/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_xlsr53/layer20/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c50488e9e8b9f83e3743a1e0aa60b31d7e3c38e45c421b06da3c734e2743a1f1
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer20/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f557d716907f5344d87c1ad00ae44e9c99276ded924802742444e4ad13ddeb2
3
+ size 3404998
results_bvcc/w2v2_large_xlsr53/layer20/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.5252037472449816
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3260 | LCC = 0.8098 | SRCC = 0.8132 ]
8
+ INFO:root:Average loss=0.3006973887483279
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2792 | LCC = 0.8349 | SRCC = 0.8337 ]
11
+ INFO:root:Average loss=0.28456209275202876
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2737 | LCC = 0.8432 | SRCC = 0.8399 ]
14
+ INFO:root:Average loss=0.2644761571517357
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2677 | LCC = 0.8417 | SRCC = 0.8395 ]
17
+ INFO:root:Average loss=0.2482989699794696
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2976 | LCC = 0.8481 | SRCC = 0.8462 ]
20
+ INFO:root:Average loss=0.23510840726204407
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2639 | LCC = 0.8520 | SRCC = 0.8486 ]
23
+ INFO:root:Average loss=0.26927776319476276
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2799 | LCC = 0.8532 | SRCC = 0.8506 ]
26
+ INFO:root:Average loss=0.22184919241147164
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2623 | LCC = 0.8512 | SRCC = 0.8503 ]
29
+ INFO:root:Average loss=0.22584270418454439
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2872 | LCC = 0.8481 | SRCC = 0.8487 ]
32
+ INFO:root:Average loss=0.213427450412359
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2583 | LCC = 0.8548 | SRCC = 0.8506 ]
35
+ INFO:root:Average loss=0.20632134740933394
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2672 | LCC = 0.8518 | SRCC = 0.8513 ]
38
+ INFO:root:Average loss=0.20528878816045248
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2751 | LCC = 0.8532 | SRCC = 0.8495 ]
41
+ INFO:root:Average loss=0.20711010006757882
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2598 | LCC = 0.8545 | SRCC = 0.8502 ]
44
+ INFO:root:Average loss=0.20164973871448102
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2799 | LCC = 0.8524 | SRCC = 0.8528 ]
47
+ INFO:root:Average loss=0.1846661360409015
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2438 | LCC = 0.8565 | SRCC = 0.8528 ]
50
+ INFO:root:Average loss=0.18137922004247323
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.3013 | LCC = 0.8507 | SRCC = 0.8487 ]
53
+ INFO:root:Average loss=0.19190689692130455
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2794 | LCC = 0.8563 | SRCC = 0.8547 ]
56
+ INFO:root:Average loss=0.18160534296662378
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2580 | LCC = 0.8580 | SRCC = 0.8546 ]
59
+ INFO:root:Average loss=0.16761686232609627
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2677 | LCC = 0.8524 | SRCC = 0.8545 ]
62
+ INFO:root:Average loss=0.17268649517343596
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2744 | LCC = 0.8553 | SRCC = 0.8557 ]
65
+ INFO:root:Average loss=0.16771187575963828
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2467 | LCC = 0.8589 | SRCC = 0.8576 ]
68
+ INFO:root:Average loss=0.16012380635127044
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2472 | LCC = 0.8602 | SRCC = 0.8587 ]
71
+ INFO:root:Average loss=0.1710203097989926
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2308 | LCC = 0.8568 | SRCC = 0.8537 ]
74
+ INFO:root:Average loss=0.15017850888081086
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2880 | LCC = 0.8546 | SRCC = 0.8551 ]
77
+ INFO:root:Average loss=0.14663071156694338
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2700 | LCC = 0.8556 | SRCC = 0.8555 ]
80
+ INFO:root:Average loss=0.14466455894020888
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2646 | LCC = 0.8584 | SRCC = 0.8580 ]
83
+ INFO:root:Average loss=0.1419999205913299
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2210 | LCC = 0.8589 | SRCC = 0.8565 ]
86
+ INFO:root:Average loss=0.14327592144791895
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2841 | LCC = 0.8573 | SRCC = 0.8544 ]
89
+ INFO:root:Average loss=0.13189556769644603
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2464 | LCC = 0.8553 | SRCC = 0.8540 ]
92
+ INFO:root:Average loss=0.12708164933018196
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2558 | LCC = 0.8581 | SRCC = 0.8559 ]
95
+ INFO:root:Average loss=0.1344688210922938
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2375 | LCC = 0.8540 | SRCC = 0.8516 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2276 | LCC = 0.8744 | SRCC = 0.8738 ]
results_bvcc/w2v2_large_xlsr53/layer22/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_xlsr53/layer22/run0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1a56f815795e931cc454630646df20dca752d5208210c5deb5516fcfe734a0c
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer22/run0/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8e5f07ead89a4d410d80619c52e4a0485ab0e3c54493edec80831b864f3770d
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/layer22/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.814239657842196
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.8270 | LCC = -0.0026 | SRCC = 0.0041 ]
8
+ INFO:root:Average loss=0.8085363232172452
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.8259 | LCC = 0.0493 | SRCC = 0.0578 ]
11
+ INFO:root:Average loss=0.8069201738406451
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.8247 | LCC = 0.1962 | SRCC = 0.1724 ]
14
+ INFO:root:Average loss=0.804259663208937
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.8243 | LCC = 0.1360 | SRCC = 0.0895 ]
17
+ INFO:root:Average loss=0.7963876479711288
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.8091 | LCC = 0.2472 | SRCC = 0.2498 ]
20
+ INFO:root:Average loss=0.7853928804397583
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.7979 | LCC = 0.2328 | SRCC = 0.2393 ]
23
+ INFO:root:Average loss=0.7715532069022839
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.7915 | LCC = 0.2385 | SRCC = 0.2449 ]
26
+ INFO:root:Average loss=0.7595658860145471
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.7796 | LCC = 0.2547 | SRCC = 0.2535 ]
29
+ INFO:root:Average loss=0.7552185222888604
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.7782 | LCC = 0.2615 | SRCC = 0.2638 ]
32
+ INFO:root:Average loss=0.7525569139382778
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.7804 | LCC = 0.2499 | SRCC = 0.2511 ]
35
+ INFO:root:Average loss=0.7490798051540668
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.7807 | LCC = 0.2506 | SRCC = 0.2477 ]
38
+ INFO:root:Average loss=0.7523076916352297
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.7763 | LCC = 0.2525 | SRCC = 0.2499 ]
41
+ INFO:root:Average loss=0.7473140374208108
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.7775 | LCC = 0.2565 | SRCC = 0.2576 ]
44
+ INFO:root:Average loss=0.7457434053604419
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.7724 | LCC = 0.2590 | SRCC = 0.2566 ]
47
+ INFO:root:Average loss=0.7413836671755865
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.7757 | LCC = 0.2561 | SRCC = 0.2580 ]
50
+ INFO:root:Average loss=0.746223298402933
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.7791 | LCC = 0.2531 | SRCC = 0.2559 ]
53
+ INFO:root:Average loss=0.7420797638404064
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.7749 | LCC = 0.2586 | SRCC = 0.2605 ]
56
+ INFO:root:Average loss=0.7429421719832298
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.7730 | LCC = 0.2617 | SRCC = 0.2634 ]
59
+ INFO:root:Average loss=0.7376499966933177
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.7722 | LCC = 0.2626 | SRCC = 0.2612 ]
62
+ INFO:root:Average loss=0.7371719991549467
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.7723 | LCC = 0.2641 | SRCC = 0.2609 ]
65
+ INFO:root:Average loss=0.7375246989421356
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.7737 | LCC = 0.2596 | SRCC = 0.2569 ]
68
+ INFO:root:Average loss=0.7357758390597808
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.7709 | LCC = 0.2627 | SRCC = 0.2593 ]
71
+ INFO:root:Average loss=0.7394503248043549
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.7721 | LCC = 0.2609 | SRCC = 0.2602 ]
74
+ INFO:root:Average loss=0.7379284046399288
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.7725 | LCC = 0.2575 | SRCC = 0.2581 ]
77
+ INFO:root:Average loss=0.7340310563643774
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.7736 | LCC = 0.2577 | SRCC = 0.2566 ]
80
+ INFO:root:Average loss=0.732508102288613
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.7724 | LCC = 0.2564 | SRCC = 0.2555 ]
83
+ INFO:root:Average loss=0.7312567891218723
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.7739 | LCC = 0.2568 | SRCC = 0.2540 ]
86
+ INFO:root:Average loss=0.7326350448987423
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.7737 | LCC = 0.2552 | SRCC = 0.2567 ]
89
+ INFO:root:Average loss=0.7306141421580926
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.7751 | LCC = 0.2552 | SRCC = 0.2572 ]
92
+ INFO:root:Average loss=0.7304448515940936
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.7763 | LCC = 0.2522 | SRCC = 0.2621 ]
95
+ INFO:root:Average loss=0.7258755343082623
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.7713 | LCC = 0.2600 | SRCC = 0.2612 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.8067 | LCC = 0.2564 | SRCC = 0.2638 ]
results_bvcc/w2v2_large_xlsr53/layer22/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_xlsr53/layer22/run1/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a4207c575fda017ff98028b9abcd06aa3d9d89765d6e50c7f88d11e2c0f6f64
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer22/run1/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13807fed47bdbbf469ba629fdbafecfe255caee85de995c0a35f4000aaa2fed2
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/layer22/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.8181307224126962
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.8383 | LCC = 0.1263 | SRCC = 0.1136 ]
8
+ INFO:root:Average loss=0.8058509589769901
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.8313 | LCC = 0.1621 | SRCC = 0.1426 ]
11
+ INFO:root:Average loss=0.8050016615635309
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.8302 | LCC = 0.1852 | SRCC = 0.1710 ]
14
+ INFO:root:Average loss=0.8022013230201526
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.8238 | LCC = 0.2060 | SRCC = 0.2061 ]
17
+ INFO:root:Average loss=0.7932518109297141
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.8095 | LCC = 0.2203 | SRCC = 0.2173 ]
20
+ INFO:root:Average loss=0.7824825889024979
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.8238 | LCC = 0.2229 | SRCC = 0.2354 ]
23
+ INFO:root:Average loss=0.7676528906210874
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.7966 | LCC = 0.2300 | SRCC = 0.2333 ]
26
+ INFO:root:Average loss=0.7637531146025046
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.7890 | LCC = 0.2374 | SRCC = 0.2388 ]
29
+ INFO:root:Average loss=0.7614126335351895
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.7893 | LCC = 0.2364 | SRCC = 0.2401 ]
32
+ INFO:root:Average loss=0.7546045344609481
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.7927 | LCC = 0.2397 | SRCC = 0.2433 ]
35
+ INFO:root:Average loss=0.7521014488660372
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.7971 | LCC = 0.2430 | SRCC = 0.2452 ]
38
+ INFO:root:Average loss=0.7542945650907663
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.8035 | LCC = 0.2403 | SRCC = 0.2467 ]
41
+ INFO:root:Average loss=0.7496093186812524
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.8264 | LCC = 0.2363 | SRCC = 0.2470 ]
44
+ INFO:root:Average loss=0.7481262401128427
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.8146 | LCC = 0.2366 | SRCC = 0.2440 ]
47
+ INFO:root:Average loss=0.7483717394180787
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.7988 | LCC = 0.2392 | SRCC = 0.2478 ]
50
+ INFO:root:Average loss=0.7470266200028933
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.7773 | LCC = 0.2472 | SRCC = 0.2479 ]
53
+ INFO:root:Average loss=0.7472251134041028
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.8218 | LCC = 0.2336 | SRCC = 0.2444 ]
56
+ INFO:root:Average loss=0.7448329543456053
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.8404 | LCC = 0.2386 | SRCC = 0.2471 ]
59
+ INFO:root:Average loss=0.7468257385950822
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.8102 | LCC = 0.2413 | SRCC = 0.2495 ]
62
+ INFO:root:Average loss=0.7398762595959198
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.7956 | LCC = 0.2506 | SRCC = 0.2513 ]
65
+ INFO:root:Average loss=0.739950743623269
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.7798 | LCC = 0.2558 | SRCC = 0.2576 ]
68
+ INFO:root:Average loss=0.74148706977184
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.7825 | LCC = 0.2547 | SRCC = 0.2602 ]
71
+ INFO:root:Average loss=0.7412474002593603
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.8015 | LCC = 0.2487 | SRCC = 0.2586 ]
74
+ INFO:root:Average loss=0.7376955036933606
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.7965 | LCC = 0.2547 | SRCC = 0.2586 ]
77
+ INFO:root:Average loss=0.7389899431130825
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.8086 | LCC = 0.2474 | SRCC = 0.2556 ]
80
+ INFO:root:Average loss=0.7341073384651771
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.7827 | LCC = 0.2564 | SRCC = 0.2639 ]
83
+ INFO:root:Average loss=0.7389515500802261
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.7884 | LCC = 0.2528 | SRCC = 0.2571 ]
86
+ INFO:root:Average loss=0.7339698744890018
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.7711 | LCC = 0.2622 | SRCC = 0.2609 ]
89
+ INFO:root:Average loss=0.737748962182265
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.7770 | LCC = 0.2527 | SRCC = 0.2605 ]
92
+ INFO:root:Average loss=0.7384612575555459
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.7779 | LCC = 0.2590 | SRCC = 0.2643 ]
95
+ INFO:root:Average loss=0.7393181075652441
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.8223 | LCC = 0.2459 | SRCC = 0.2583 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.7996 | LCC = 0.2422 | SRCC = 0.2504 ]
results_bvcc/w2v2_large_xlsr53/layer22/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_xlsr53/layer22/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:600e13b24b4df5a9bad0859914464388871330115787498ebd9cf9ef13c194bc
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer22/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc6b697fc358ef048a7d3407572dcc3c9f3ee6f595b36f5cc664485cc6f7138c
3
+ size 3404998
results_bvcc/w2v2_large_xlsr53/layer22/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.8170791971377838
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.8325 | LCC = 0.1026 | SRCC = 0.1114 ]
8
+ INFO:root:Average loss=0.8104762167502673
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.8263 | LCC = -0.0209 | SRCC = -0.0027 ]
11
+ INFO:root:Average loss=0.8071949627154913
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.8266 | LCC = 0.0121 | SRCC = 0.0360 ]
14
+ INFO:root:Average loss=0.8096001698420598
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.8256 | LCC = 0.0345 | SRCC = 0.0465 ]
17
+ INFO:root:Average loss=0.8068318619177892
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.8247 | LCC = 0.0963 | SRCC = 0.1003 ]
20
+ INFO:root:Average loss=0.8068208495775858
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.8289 | LCC = 0.1201 | SRCC = 0.1209 ]
23
+ INFO:root:Average loss=0.8082066774368286
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.8299 | LCC = 0.1215 | SRCC = 0.1197 ]
26
+ INFO:root:Average loss=0.7999909252692492
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.8185 | LCC = 0.2234 | SRCC = 0.2272 ]
29
+ INFO:root:Average loss=0.7974462914161193
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.8092 | LCC = 0.2301 | SRCC = 0.2401 ]
32
+ INFO:root:Average loss=0.7786302451904004
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.8018 | LCC = 0.2356 | SRCC = 0.2532 ]
35
+ INFO:root:Average loss=0.7637408131208175
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.8020 | LCC = 0.2301 | SRCC = 0.2432 ]
38
+ INFO:root:Average loss=0.7688980881984417
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.7847 | LCC = 0.2382 | SRCC = 0.2453 ]
41
+ INFO:root:Average loss=0.7564181311008258
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.7874 | LCC = 0.2371 | SRCC = 0.2423 ]
44
+ INFO:root:Average loss=0.7545683070635184
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.8091 | LCC = 0.2361 | SRCC = 0.2466 ]
47
+ INFO:root:Average loss=0.7512389161647894
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.7793 | LCC = 0.2450 | SRCC = 0.2449 ]
50
+ INFO:root:Average loss=0.7586557078055847
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.7864 | LCC = 0.2386 | SRCC = 0.2431 ]
53
+ INFO:root:Average loss=0.7566420504680047
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.7858 | LCC = 0.2423 | SRCC = 0.2426 ]
56
+ INFO:root:Average loss=0.7498466303715339
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.7781 | LCC = 0.2444 | SRCC = 0.2435 ]
59
+ INFO:root:Average loss=0.7462894381620945
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.8138 | LCC = 0.2368 | SRCC = 0.2465 ]
62
+ INFO:root:Average loss=0.7527802273249015
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.7938 | LCC = 0.2362 | SRCC = 0.2415 ]
65
+ INFO:root:Average loss=0.7473375663543359
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.7803 | LCC = 0.2437 | SRCC = 0.2459 ]
68
+ INFO:root:Average loss=0.7387752280785487
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.7824 | LCC = 0.2478 | SRCC = 0.2484 ]
71
+ INFO:root:Average loss=0.7506021536313571
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.7798 | LCC = 0.2467 | SRCC = 0.2469 ]
74
+ INFO:root:Average loss=0.742162606655023
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.7770 | LCC = 0.2476 | SRCC = 0.2459 ]
77
+ INFO:root:Average loss=0.7410742491483688
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.7751 | LCC = 0.2503 | SRCC = 0.2458 ]
80
+ INFO:root:Average loss=0.7445911574058044
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.7795 | LCC = 0.2432 | SRCC = 0.2463 ]
83
+ INFO:root:Average loss=0.7379241952529321
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.7901 | LCC = 0.2368 | SRCC = 0.2378 ]
86
+ INFO:root:Average loss=0.7360416750113169
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.7746 | LCC = 0.2503 | SRCC = 0.2480 ]
89
+ INFO:root:Average loss=0.7346521952213385
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.7962 | LCC = 0.2415 | SRCC = 0.2478 ]
92
+ INFO:root:Average loss=0.7360071792052343
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.7771 | LCC = 0.2491 | SRCC = 0.2513 ]
95
+ INFO:root:Average loss=0.7348292752718314
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.7890 | LCC = 0.2410 | SRCC = 0.2432 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.7907 | LCC = 0.2665 | SRCC = 0.2687 ]
results_bvcc/w2v2_large_xlsr53/layer22/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_xlsr53/layer22/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d90223c11566e5b9366d1f191619706c6868e92086587ede254eea29795319c
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer22/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5055a81939fd491e91cffaa54a4a2efad6ceca33ada7ff52dff767a619970f25
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/layer22/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.8139951924482981
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.8538 | LCC = 0.0628 | SRCC = 0.0433 ]
8
+ INFO:root:Average loss=0.8084817964297074
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.8346 | LCC = 0.1495 | SRCC = 0.1408 ]
11
+ INFO:root:Average loss=0.8109443149505517
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.8399 | LCC = 0.1735 | SRCC = 0.1573 ]
14
+ INFO:root:Average loss=0.8042151973797724
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.8378 | LCC = 0.2081 | SRCC = 0.2045 ]
17
+ INFO:root:Average loss=0.8034162750610938
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.8440 | LCC = 0.2189 | SRCC = 0.2294 ]
20
+ INFO:root:Average loss=0.807106227446825
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.8303 | LCC = 0.2063 | SRCC = 0.2054 ]
23
+ INFO:root:Average loss=0.7967051993577908
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.8262 | LCC = 0.2212 | SRCC = 0.2231 ]
26
+ INFO:root:Average loss=0.7873269961430476
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.8369 | LCC = 0.2284 | SRCC = 0.2433 ]
29
+ INFO:root:Average loss=0.7755554616451263
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.8117 | LCC = 0.2389 | SRCC = 0.2573 ]
32
+ INFO:root:Average loss=0.7619453622744634
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.7796 | LCC = 0.2543 | SRCC = 0.2549 ]
35
+ INFO:root:Average loss=0.7669727420195555
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.7817 | LCC = 0.2582 | SRCC = 0.2596 ]
38
+ INFO:root:Average loss=0.7573283819051889
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.8115 | LCC = 0.2542 | SRCC = 0.2628 ]
41
+ INFO:root:Average loss=0.753503374564342
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.7973 | LCC = 0.2575 | SRCC = 0.2668 ]
44
+ INFO:root:Average loss=0.7453485207679944
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.8421 | LCC = 0.2580 | SRCC = 0.2705 ]
47
+ INFO:root:Average loss=0.7502139508724213
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.7675 | LCC = 0.2719 | SRCC = 0.2685 ]
50
+ INFO:root:Average loss=0.7436514829213803
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.7981 | LCC = 0.2540 | SRCC = 0.2611 ]
53
+ INFO:root:Average loss=0.7450853158266116
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.7642 | LCC = 0.2791 | SRCC = 0.2766 ]
56
+ INFO:root:Average loss=0.7439048061004052
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.7791 | LCC = 0.2748 | SRCC = 0.2763 ]
59
+ INFO:root:Average loss=0.7414584794105628
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.7793 | LCC = 0.2727 | SRCC = 0.2757 ]
62
+ INFO:root:Average loss=0.7410066070464941
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.7671 | LCC = 0.2795 | SRCC = 0.2782 ]
65
+ INFO:root:Average loss=0.7383616111981564
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.7835 | LCC = 0.2746 | SRCC = 0.2757 ]
68
+ INFO:root:Average loss=0.7370491738502796
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.8018 | LCC = 0.2653 | SRCC = 0.2673 ]
71
+ INFO:root:Average loss=0.7324779484516535
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.7692 | LCC = 0.2761 | SRCC = 0.2786 ]
74
+ INFO:root:Average loss=0.740232072579555
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.7750 | LCC = 0.2804 | SRCC = 0.2792 ]
77
+ INFO:root:Average loss=0.7392700532307992
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.7744 | LCC = 0.2807 | SRCC = 0.2760 ]
80
+ INFO:root:Average loss=0.7341036876806846
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.7650 | LCC = 0.2814 | SRCC = 0.2765 ]
83
+ INFO:root:Average loss=0.7312420346798041
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.7766 | LCC = 0.2829 | SRCC = 0.2840 ]
86
+ INFO:root:Average loss=0.7305280428666335
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.7855 | LCC = 0.2772 | SRCC = 0.2782 ]
89
+ INFO:root:Average loss=0.7277241028272189
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.7805 | LCC = 0.2800 | SRCC = 0.2774 ]
92
+ INFO:root:Average loss=0.7310245992281498
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.7570 | LCC = 0.2933 | SRCC = 0.2889 ]
95
+ INFO:root:Average loss=0.7278549052201785
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.7954 | LCC = 0.2802 | SRCC = 0.2838 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.7960 | LCC = 0.2681 | SRCC = 0.2799 ]
results_bvcc/w2v2_large_xlsr53/layer22/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_xlsr53/layer22/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9bb001e0a7e711f1bb9275c034935d862ccdc20710f9458f8eefba1a22f9319
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer22/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:439a309d13eef5a0952916c0c3681ace23ef98e55fffd31d55e0902364494747
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/layer22/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.8078734744817783
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.8438 | LCC = 0.1551 | SRCC = 0.1338 ]
8
+ INFO:root:Average loss=0.8061453310342935
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.8527 | LCC = 0.1858 | SRCC = 0.1696 ]
11
+ INFO:root:Average loss=0.7980850124970461
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.8782 | LCC = 0.2039 | SRCC = 0.2082 ]
14
+ INFO:root:Average loss=0.7904722224443387
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.8657 | LCC = 0.2205 | SRCC = 0.2218 ]
17
+ INFO:root:Average loss=0.7704049562796568
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.8042 | LCC = 0.2416 | SRCC = 0.2469 ]
20
+ INFO:root:Average loss=0.7644153917447115
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.8250 | LCC = 0.2385 | SRCC = 0.2437 ]
23
+ INFO:root:Average loss=0.7613810132711362
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.8584 | LCC = 0.2353 | SRCC = 0.2405 ]
26
+ INFO:root:Average loss=0.7635564146897732
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.8602 | LCC = 0.2334 | SRCC = 0.2399 ]
29
+ INFO:root:Average loss=0.7606048110203866
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.7999 | LCC = 0.2399 | SRCC = 0.2370 ]
32
+ INFO:root:Average loss=0.7518155513665615
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.7966 | LCC = 0.2453 | SRCC = 0.2419 ]
35
+ INFO:root:Average loss=0.7531586511012835
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.8018 | LCC = 0.2439 | SRCC = 0.2402 ]
38
+ INFO:root:Average loss=0.7529040429836664
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.8175 | LCC = 0.2530 | SRCC = 0.2528 ]
41
+ INFO:root:Average loss=0.7498894028174572
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.8310 | LCC = 0.2520 | SRCC = 0.2505 ]
44
+ INFO:root:Average loss=0.7508819340131222
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.8223 | LCC = 0.2518 | SRCC = 0.2502 ]
47
+ INFO:root:Average loss=0.7533572163337316
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.8383 | LCC = 0.2479 | SRCC = 0.2476 ]
50
+ INFO:root:Average loss=0.7443092568562582
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.8035 | LCC = 0.2544 | SRCC = 0.2475 ]
53
+ INFO:root:Average loss=0.7448368473694875
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.8218 | LCC = 0.2568 | SRCC = 0.2515 ]
56
+ INFO:root:Average loss=0.748010686574838
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.8373 | LCC = 0.2558 | SRCC = 0.2524 ]
59
+ INFO:root:Average loss=0.7410632696671363
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.8248 | LCC = 0.2561 | SRCC = 0.2532 ]
62
+ INFO:root:Average loss=0.7445670266946157
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.8269 | LCC = 0.2542 | SRCC = 0.2514 ]
65
+ INFO:root:Average loss=0.74035588823832
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.8644 | LCC = 0.2530 | SRCC = 0.2545 ]
68
+ INFO:root:Average loss=0.7474632763709778
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.8147 | LCC = 0.2602 | SRCC = 0.2546 ]
71
+ INFO:root:Average loss=0.7406386037667593
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.8054 | LCC = 0.2618 | SRCC = 0.2536 ]
74
+ INFO:root:Average loss=0.7379722109971902
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.8150 | LCC = 0.2635 | SRCC = 0.2614 ]
77
+ INFO:root:Average loss=0.7341487545233506
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.8073 | LCC = 0.2630 | SRCC = 0.2580 ]
80
+ INFO:root:Average loss=0.7340960743335577
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.7963 | LCC = 0.2616 | SRCC = 0.2544 ]
83
+ INFO:root:Average loss=0.7333975468690579
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.8081 | LCC = 0.2641 | SRCC = 0.2592 ]
86
+ INFO:root:Average loss=0.7372842583900843
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.8026 | LCC = 0.2604 | SRCC = 0.2548 ]
89
+ INFO:root:Average loss=0.7322733646784073
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.8087 | LCC = 0.2649 | SRCC = 0.2605 ]
92
+ INFO:root:Average loss=0.7382883513585116
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.7805 | LCC = 0.2660 | SRCC = 0.2607 ]
95
+ INFO:root:Average loss=0.7296412785847982
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.7854 | LCC = 0.2664 | SRCC = 0.2601 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.7988 | LCC = 0.2486 | SRCC = 0.2584 ]
results_bvcc/w2v2_large_xlsr53/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_large_xlsr53/layer4/run0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d17ddd1fc840a4c8efb2d94b5c79a8182ea133d0da68d1c78d08949e45d54f9
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer4/run0/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c77faa9c79a76bc831692b5f9b595ac4ccb9406a66d60eb59c0a3cdc68cdec7d
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/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.547156467460669
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3832 | LCC = 0.7870 | SRCC = 0.7937 ]
8
+ INFO:root:Average loss=0.29149841994811326
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3145 | LCC = 0.8253 | SRCC = 0.8260 ]
11
+ INFO:root:Average loss=0.24849581298155662
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2810 | LCC = 0.8434 | SRCC = 0.8447 ]
14
+ INFO:root:Average loss=0.22738109395290032
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2621 | LCC = 0.8573 | SRCC = 0.8536 ]
17
+ INFO:root:Average loss=0.21490304286663348
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2838 | LCC = 0.8643 | SRCC = 0.8606 ]
20
+ INFO:root:Average loss=0.20760062088569006
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2279 | LCC = 0.8673 | SRCC = 0.8642 ]
23
+ INFO:root:Average loss=0.20366449520373955
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2377 | LCC = 0.8735 | SRCC = 0.8702 ]
26
+ INFO:root:Average loss=0.17576329687084907
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2163 | LCC = 0.8735 | SRCC = 0.8715 ]
29
+ INFO:root:Average loss=0.17690044412246117
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2238 | LCC = 0.8779 | SRCC = 0.8757 ]
32
+ INFO:root:Average loss=0.1625153881808122
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2058 | LCC = 0.8757 | SRCC = 0.8747 ]
35
+ INFO:root:Average loss=0.16646776797297674
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2525 | LCC = 0.8692 | SRCC = 0.8712 ]
38
+ INFO:root:Average loss=0.16911816062071386
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2083 | LCC = 0.8794 | SRCC = 0.8774 ]
41
+ INFO:root:Average loss=0.14906550905643365
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2191 | LCC = 0.8795 | SRCC = 0.8771 ]
44
+ INFO:root:Average loss=0.14371993793891027
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2117 | LCC = 0.8791 | SRCC = 0.8769 ]
47
+ INFO:root:Average loss=0.1384089978841635
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2030 | LCC = 0.8787 | SRCC = 0.8776 ]
50
+ INFO:root:Average loss=0.12990750276889557
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2243 | LCC = 0.8789 | SRCC = 0.8775 ]
53
+ INFO:root:Average loss=0.13416247537884957
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.1994 | LCC = 0.8776 | SRCC = 0.8764 ]
56
+ INFO:root:Average loss=0.1280193323126206
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2107 | LCC = 0.8757 | SRCC = 0.8748 ]
59
+ INFO:root:Average loss=0.11507375280444439
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2014 | LCC = 0.8787 | SRCC = 0.8778 ]
62
+ INFO:root:Average loss=0.11986773652143967
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2211 | LCC = 0.8760 | SRCC = 0.8761 ]
65
+ INFO:root:Average loss=0.10640933001652741
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2026 | LCC = 0.8763 | SRCC = 0.8763 ]
68
+ INFO:root:Average loss=0.10712833387347367
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.1972 | LCC = 0.8812 | SRCC = 0.8802 ]
71
+ INFO:root:Average loss=0.10519853802636647
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2054 | LCC = 0.8747 | SRCC = 0.8757 ]
74
+ INFO:root:Average loss=0.09490100838817082
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2048 | LCC = 0.8798 | SRCC = 0.8781 ]
77
+ INFO:root:Average loss=0.09653763964963265
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2104 | LCC = 0.8747 | SRCC = 0.8743 ]
80
+ INFO:root:Average loss=0.09158640760832872
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2154 | LCC = 0.8728 | SRCC = 0.8724 ]
83
+ INFO:root:Average loss=0.08465714289400822
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2098 | LCC = 0.8769 | SRCC = 0.8759 ]
86
+ INFO:root:Average loss=0.0891813850746705
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2106 | LCC = 0.8787 | SRCC = 0.8780 ]
89
+ INFO:root:Average loss=0.09102073861047244
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2081 | LCC = 0.8761 | SRCC = 0.8754 ]
92
+ INFO:root:Average loss=0.08620971317092578
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2109 | LCC = 0.8755 | SRCC = 0.8746 ]
95
+ INFO:root:Average loss=0.07545238857467969
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2072 | LCC = 0.8736 | SRCC = 0.8727 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2096 | LCC = 0.8749 | SRCC = 0.8742 ]
results_bvcc/w2v2_large_xlsr53/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_large_xlsr53/layer4/run1/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf2cd0da48458a8dad5a0439c598dc91722a10bd72eb1293d122b1b3df500b43
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer4/run1/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce6fd6a252785a6f331f9946f5b53c13f73e0ee5bc262d421079252183450b33
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/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.5605818294943907
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3382 | LCC = 0.7913 | SRCC = 0.7967 ]
8
+ INFO:root:Average loss=0.30025573762563557
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3533 | LCC = 0.8229 | SRCC = 0.8292 ]
11
+ INFO:root:Average loss=0.2813934796513655
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2893 | LCC = 0.8451 | SRCC = 0.8444 ]
14
+ INFO:root:Average loss=0.2493511744034596
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2820 | LCC = 0.8520 | SRCC = 0.8513 ]
17
+ INFO:root:Average loss=0.22732358005566475
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2426 | LCC = 0.8572 | SRCC = 0.8562 ]
20
+ INFO:root:Average loss=0.2069613333695974
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2915 | LCC = 0.8623 | SRCC = 0.8646 ]
23
+ INFO:root:Average loss=0.19090187750183618
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2433 | LCC = 0.8679 | SRCC = 0.8667 ]
26
+ INFO:root:Average loss=0.1918311948195482
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2164 | LCC = 0.8719 | SRCC = 0.8712 ]
29
+ INFO:root:Average loss=0.17321445105167535
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2449 | LCC = 0.8719 | SRCC = 0.8702 ]
32
+ INFO:root:Average loss=0.16294093773915216
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2188 | LCC = 0.8742 | SRCC = 0.8726 ]
35
+ INFO:root:Average loss=0.16203909204938474
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2208 | LCC = 0.8726 | SRCC = 0.8716 ]
38
+ INFO:root:Average loss=0.15740516571662364
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2292 | LCC = 0.8721 | SRCC = 0.8704 ]
41
+ INFO:root:Average loss=0.15466257461752647
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2270 | LCC = 0.8726 | SRCC = 0.8720 ]
44
+ INFO:root:Average loss=0.14520136543955559
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2052 | LCC = 0.8769 | SRCC = 0.8781 ]
47
+ INFO:root:Average loss=0.1345938279842719
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2260 | LCC = 0.8744 | SRCC = 0.8731 ]
50
+ INFO:root:Average loss=0.13297921266311255
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2284 | LCC = 0.8768 | SRCC = 0.8740 ]
53
+ INFO:root:Average loss=0.1225662044225595
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2110 | LCC = 0.8779 | SRCC = 0.8772 ]
56
+ INFO:root:Average loss=0.11552520105854058
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2273 | LCC = 0.8745 | SRCC = 0.8736 ]
59
+ INFO:root:Average loss=0.12532876174037272
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2443 | LCC = 0.8745 | SRCC = 0.8753 ]
62
+ INFO:root:Average loss=0.11093265993090776
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2307 | LCC = 0.8756 | SRCC = 0.8744 ]
65
+ INFO:root:Average loss=0.10863688277701537
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2385 | LCC = 0.8771 | SRCC = 0.8765 ]
68
+ INFO:root:Average loss=0.10018666957815488
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2663 | LCC = 0.8731 | SRCC = 0.8738 ]
71
+ INFO:root:Average loss=0.10150594340685086
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2035 | LCC = 0.8754 | SRCC = 0.8743 ]
74
+ INFO:root:Average loss=0.09620927025874455
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2666 | LCC = 0.8717 | SRCC = 0.8743 ]
77
+ INFO:root:Average loss=0.08765348242834592
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2430 | LCC = 0.8761 | SRCC = 0.8768 ]
80
+ INFO:root:Average loss=0.09197935820198976
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2052 | LCC = 0.8761 | SRCC = 0.8753 ]
83
+ INFO:root:Average loss=0.08101947677250092
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2142 | LCC = 0.8765 | SRCC = 0.8780 ]
86
+ INFO:root:Average loss=0.07929057675676468
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2166 | LCC = 0.8781 | SRCC = 0.8765 ]
89
+ INFO:root:Average loss=0.08396360980203518
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2919 | LCC = 0.8682 | SRCC = 0.8720 ]
92
+ INFO:root:Average loss=0.074722120204033
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2033 | LCC = 0.8764 | SRCC = 0.8761 ]
95
+ INFO:root:Average loss=0.0711737318107715
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2367 | LCC = 0.8755 | SRCC = 0.8754 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2166 | LCC = 0.8761 | SRCC = 0.8773 ]
results_bvcc/w2v2_large_xlsr53/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_large_xlsr53/layer4/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4988351cc534c94f99f7559f933edb0081f47cafa055d2a3e17c09d9ed472b3a
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer4/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ababe9b3184cb9bc8887520d5ba3bf14d8176e8d09fbc642ac6cc11c9daddcd0
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/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.4532563615685854
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3356 | LCC = 0.7989 | SRCC = 0.8013 ]
8
+ INFO:root:Average loss=0.27491417641823107
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2828 | LCC = 0.8347 | SRCC = 0.8370 ]
11
+ INFO:root:Average loss=0.2513206410102355
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2710 | LCC = 0.8484 | SRCC = 0.8482 ]
14
+ INFO:root:Average loss=0.2428954717440483
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2799 | LCC = 0.8471 | SRCC = 0.8524 ]
17
+ INFO:root:Average loss=0.21514941904789361
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2534 | LCC = 0.8650 | SRCC = 0.8626 ]
20
+ INFO:root:Average loss=0.2048763797069207
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2392 | LCC = 0.8661 | SRCC = 0.8668 ]
23
+ INFO:root:Average loss=0.19460664336115885
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2166 | LCC = 0.8683 | SRCC = 0.8700 ]
26
+ INFO:root:Average loss=0.18061789364004746
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2214 | LCC = 0.8693 | SRCC = 0.8684 ]
29
+ INFO:root:Average loss=0.18510188811864609
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2186 | LCC = 0.8743 | SRCC = 0.8720 ]
32
+ INFO:root:Average loss=0.1711409091949463
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2395 | LCC = 0.8728 | SRCC = 0.8728 ]
35
+ INFO:root:Average loss=0.1823367731502423
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2216 | LCC = 0.8695 | SRCC = 0.8729 ]
38
+ INFO:root:Average loss=0.15918540228635836
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2110 | LCC = 0.8734 | SRCC = 0.8718 ]
41
+ INFO:root:Average loss=0.15191630302713469
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2003 | LCC = 0.8755 | SRCC = 0.8753 ]
44
+ INFO:root:Average loss=0.15265449738273254
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2462 | LCC = 0.8677 | SRCC = 0.8726 ]
47
+ INFO:root:Average loss=0.1382217956468081
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2130 | LCC = 0.8727 | SRCC = 0.8747 ]
50
+ INFO:root:Average loss=0.14303912518498224
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2350 | LCC = 0.8722 | SRCC = 0.8728 ]
53
+ INFO:root:Average loss=0.13195646449159354
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2163 | LCC = 0.8779 | SRCC = 0.8764 ]
56
+ INFO:root:Average loss=0.12995244027712408
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2148 | LCC = 0.8735 | SRCC = 0.8752 ]
59
+ INFO:root:Average loss=0.12340931288706951
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2682 | LCC = 0.8657 | SRCC = 0.8737 ]
62
+ INFO:root:Average loss=0.11715158571799596
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2107 | LCC = 0.8726 | SRCC = 0.8745 ]
65
+ INFO:root:Average loss=0.1106030118580048
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2178 | LCC = 0.8728 | SRCC = 0.8728 ]
68
+ INFO:root:Average loss=0.11555583459826615
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2155 | LCC = 0.8780 | SRCC = 0.8771 ]
71
+ INFO:root:Average loss=0.10848726771580867
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2078 | LCC = 0.8758 | SRCC = 0.8753 ]
74
+ INFO:root:Average loss=0.09563501962484458
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2425 | LCC = 0.8681 | SRCC = 0.8676 ]
77
+ INFO:root:Average loss=0.09557310367623965
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2283 | LCC = 0.8726 | SRCC = 0.8758 ]
80
+ INFO:root:Average loss=0.09602287803322841
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.1953 | LCC = 0.8783 | SRCC = 0.8786 ]
83
+ INFO:root:Average loss=0.10340599371836735
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2294 | LCC = 0.8741 | SRCC = 0.8754 ]
86
+ INFO:root:Average loss=0.08820458687841892
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2308 | LCC = 0.8733 | SRCC = 0.8737 ]
89
+ INFO:root:Average loss=0.08390403567598416
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2481 | LCC = 0.8685 | SRCC = 0.8749 ]
92
+ INFO:root:Average loss=0.0964965601093494
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.1953 | LCC = 0.8769 | SRCC = 0.8774 ]
95
+ INFO:root:Average loss=0.08484634895546314
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2053 | LCC = 0.8762 | SRCC = 0.8762 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2247 | LCC = 0.8717 | SRCC = 0.8723 ]
results_bvcc/w2v2_large_xlsr53/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_large_xlsr53/layer4/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:727657e5e90f3d64442de165ff4ede62d7b7457f843b30dba402b9401a31d0b3
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer4/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1f800e6f5de54cefe7b88d3f824da19de7ff6faf367c840ec7f221ca3160784
3
+ size 3405062
results_bvcc/w2v2_large_xlsr53/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.4696754650809826
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3065 | LCC = 0.8023 | SRCC = 0.8093 ]
8
+ INFO:root:Average loss=0.2954771388799716
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2991 | LCC = 0.8411 | SRCC = 0.8407 ]
11
+ INFO:root:Average loss=0.28332846611738205
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2869 | LCC = 0.8463 | SRCC = 0.8464 ]
14
+ INFO:root:Average loss=0.23383696663838166
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2663 | LCC = 0.8586 | SRCC = 0.8569 ]
17
+ INFO:root:Average loss=0.21505269962243545
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2557 | LCC = 0.8593 | SRCC = 0.8594 ]
20
+ INFO:root:Average loss=0.21830975111478415
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2455 | LCC = 0.8675 | SRCC = 0.8651 ]
23
+ INFO:root:Average loss=0.1987616410240149
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2317 | LCC = 0.8693 | SRCC = 0.8681 ]
26
+ INFO:root:Average loss=0.19079558522655413
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2670 | LCC = 0.8720 | SRCC = 0.8704 ]
29
+ INFO:root:Average loss=0.18426570716576698
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2334 | LCC = 0.8724 | SRCC = 0.8717 ]
32
+ INFO:root:Average loss=0.17354323495274934
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2210 | LCC = 0.8753 | SRCC = 0.8748 ]
35
+ INFO:root:Average loss=0.18379609582897943
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2884 | LCC = 0.8663 | SRCC = 0.8725 ]
38
+ INFO:root:Average loss=0.1759594639715476
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2341 | LCC = 0.8718 | SRCC = 0.8731 ]
41
+ INFO:root:Average loss=0.16178548221404737
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2050 | LCC = 0.8786 | SRCC = 0.8777 ]
44
+ INFO:root:Average loss=0.14725517749022216
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2202 | LCC = 0.8749 | SRCC = 0.8736 ]
47
+ INFO:root:Average loss=0.1469860810499925
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2186 | LCC = 0.8779 | SRCC = 0.8773 ]
50
+ INFO:root:Average loss=0.1449679702711411
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.1947 | LCC = 0.8786 | SRCC = 0.8785 ]
53
+ INFO:root:Average loss=0.13312828435729712
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2659 | LCC = 0.8658 | SRCC = 0.8687 ]
56
+ INFO:root:Average loss=0.1286121530410571
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2298 | LCC = 0.8746 | SRCC = 0.8741 ]
59
+ INFO:root:Average loss=0.12523656710982323
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2136 | LCC = 0.8763 | SRCC = 0.8749 ]
62
+ INFO:root:Average loss=0.12012366854991668
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2106 | LCC = 0.8773 | SRCC = 0.8772 ]
65
+ INFO:root:Average loss=0.11537392714466804
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2178 | LCC = 0.8739 | SRCC = 0.8743 ]
68
+ INFO:root:Average loss=0.10499710245774342
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2007 | LCC = 0.8805 | SRCC = 0.8794 ]
71
+ INFO:root:Average loss=0.10581345703357305
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2155 | LCC = 0.8743 | SRCC = 0.8756 ]
74
+ INFO:root:Average loss=0.10889474092385708
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2006 | LCC = 0.8767 | SRCC = 0.8754 ]
77
+ INFO:root:Average loss=0.1006790762528395
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2121 | LCC = 0.8718 | SRCC = 0.8693 ]
80
+ INFO:root:Average loss=0.09053596796897742
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2317 | LCC = 0.8743 | SRCC = 0.8740 ]
83
+ INFO:root:Average loss=0.09128491265269426
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2291 | LCC = 0.8756 | SRCC = 0.8734 ]
86
+ INFO:root:Average loss=0.08415590680371492
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2174 | LCC = 0.8759 | SRCC = 0.8760 ]
89
+ INFO:root:Average loss=0.08523786067962646
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2122 | LCC = 0.8769 | SRCC = 0.8763 ]
92
+ INFO:root:Average loss=0.08090076237343825
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2105 | LCC = 0.8771 | SRCC = 0.8764 ]
95
+ INFO:root:Average loss=0.0761953240308242
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2198 | LCC = 0.8731 | SRCC = 0.8756 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2211 | LCC = 0.8703 | SRCC = 0.8700 ]
results_bvcc/w2v2_large_xlsr53/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_large_xlsr53/layer4/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f32c3616db4d0fbebd275bee428cbbddb517f648dd559fd48498d502ae6040be
3
+ size 3403339
results_bvcc/w2v2_large_xlsr53/layer4/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:904fa759cb4ec83f126ca39d143affdeea20223e7a8c03010188a94ce91d7de4
3
+ size 3405062