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

Upload SSL Layer MOS checkpoints (part 5)

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_2epoch/layer7/run2/model.pt +3 -0
  2. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model_best.pt +3 -0
  3. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/train.log +99 -0
  4. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/config.gin +31 -0
  5. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/model.pt +3 -0
  6. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/model_best.pt +3 -0
  7. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/train.log +99 -0
  8. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/config.gin +31 -0
  9. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/model.pt +3 -0
  10. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/model_best.pt +3 -0
  11. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/train.log +99 -0
  12. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/config.gin +31 -0
  13. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/model.pt +3 -0
  14. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/model_best.pt +3 -0
  15. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/train.log +99 -0
  16. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/config.gin +31 -0
  17. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/model.pt +3 -0
  18. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/model_best.pt +3 -0
  19. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/train.log +99 -0
  20. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/config.gin +31 -0
  21. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/model.pt +3 -0
  22. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/model_best.pt +3 -0
  23. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/train.log +99 -0
  24. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/config.gin +31 -0
  25. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/model.pt +3 -0
  26. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/model_best.pt +3 -0
  27. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/train.log +99 -0
  28. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/config.gin +31 -0
  29. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/model.pt +3 -0
  30. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/model_best.pt +3 -0
  31. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/train.log +99 -0
  32. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/config.gin +31 -0
  33. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/model.pt +3 -0
  34. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/model_best.pt +3 -0
  35. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/train.log +99 -0
  36. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/config.gin +31 -0
  37. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/model.pt +3 -0
  38. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/model_best.pt +3 -0
  39. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/train.log +99 -0
  40. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/config.gin +31 -0
  41. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/model.pt +3 -0
  42. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/model_best.pt +3 -0
  43. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/train.log +99 -0
  44. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/config.gin +31 -0
  45. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/model.pt +3 -0
  46. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/model_best.pt +3 -0
  47. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/train.log +99 -0
  48. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/config.gin +31 -0
  49. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/model.pt +3 -0
  50. results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/model_best.pt +3 -0
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab412a40c6a2a7fb78b6ba80e6a890b9b5dc58db1ebf1b0ea16c266fa97849e5
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7698ab5576e7ac936a82b6fbf0546adbc3155970ce8bf6022014b940076b93e5
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/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.4333679804053062
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3561 | LCC = 0.8064 | SRCC = 0.8111 ]
8
+ INFO:root:Average loss=0.27634325871864956
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3084 | LCC = 0.8394 | SRCC = 0.8386 ]
11
+ INFO:root:Average loss=0.2417834808047001
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3485 | LCC = 0.8483 | SRCC = 0.8507 ]
14
+ INFO:root:Average loss=0.2363869091257071
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2711 | LCC = 0.8551 | SRCC = 0.8538 ]
17
+ INFO:root:Average loss=0.21763800313839546
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2544 | LCC = 0.8584 | SRCC = 0.8584 ]
20
+ INFO:root:Average loss=0.19965359090994567
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.3070 | LCC = 0.8598 | SRCC = 0.8588 ]
23
+ INFO:root:Average loss=0.18454301662934133
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2501 | LCC = 0.8572 | SRCC = 0.8555 ]
26
+ INFO:root:Average loss=0.17260222862928343
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2483 | LCC = 0.8597 | SRCC = 0.8598 ]
29
+ INFO:root:Average loss=0.16166870964643282
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2490 | LCC = 0.8598 | SRCC = 0.8600 ]
32
+ INFO:root:Average loss=0.14926960261968467
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2632 | LCC = 0.8607 | SRCC = 0.8602 ]
35
+ INFO:root:Average loss=0.14366154783429244
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2406 | LCC = 0.8575 | SRCC = 0.8566 ]
38
+ INFO:root:Average loss=0.13270336131636912
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2480 | LCC = 0.8566 | SRCC = 0.8561 ]
41
+ INFO:root:Average loss=0.12664856838110167
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2659 | LCC = 0.8546 | SRCC = 0.8570 ]
44
+ INFO:root:Average loss=0.11981937614006874
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2470 | LCC = 0.8519 | SRCC = 0.8529 ]
47
+ INFO:root:Average loss=0.1176408742291805
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2629 | LCC = 0.8542 | SRCC = 0.8551 ]
50
+ INFO:root:Average loss=0.10629363157428227
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2790 | LCC = 0.8513 | SRCC = 0.8532 ]
53
+ INFO:root:Average loss=0.09905539023188445
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2683 | LCC = 0.8479 | SRCC = 0.8480 ]
56
+ INFO:root:Average loss=0.09355268398156533
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2807 | LCC = 0.8432 | SRCC = 0.8459 ]
59
+ INFO:root:Average loss=0.0945625029122218
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2507 | LCC = 0.8487 | SRCC = 0.8497 ]
62
+ INFO:root:Average loss=0.0855995132946051
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2653 | LCC = 0.8466 | SRCC = 0.8466 ]
65
+ INFO:root:Average loss=0.08877791110903789
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2444 | LCC = 0.8511 | SRCC = 0.8523 ]
68
+ INFO:root:Average loss=0.08188509148282883
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2533 | LCC = 0.8484 | SRCC = 0.8497 ]
71
+ INFO:root:Average loss=0.07404621523351242
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2624 | LCC = 0.8462 | SRCC = 0.8473 ]
74
+ INFO:root:Average loss=0.07226133031340745
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2529 | LCC = 0.8506 | SRCC = 0.8515 ]
77
+ INFO:root:Average loss=0.06969021642819428
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2734 | LCC = 0.8433 | SRCC = 0.8429 ]
80
+ INFO:root:Average loss=0.0658117977854533
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2387 | LCC = 0.8498 | SRCC = 0.8512 ]
83
+ INFO:root:Average loss=0.06294814812449309
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2505 | LCC = 0.8490 | SRCC = 0.8490 ]
86
+ INFO:root:Average loss=0.05886615440249443
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2603 | LCC = 0.8470 | SRCC = 0.8474 ]
89
+ INFO:root:Average loss=0.06502897827289043
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2498 | LCC = 0.8478 | SRCC = 0.8487 ]
92
+ INFO:root:Average loss=0.05716708899499514
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2474 | LCC = 0.8491 | SRCC = 0.8506 ]
95
+ INFO:root:Average loss=0.05529519208730795
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2509 | LCC = 0.8480 | SRCC = 0.8496 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2618 | LCC = 0.8545 | SRCC = 0.8545 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/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_2epoch/layer7/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c044c654a235ea22bca38526a79abc04b31e7b75fec343bedeefdf651b13165
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:799fe32dda6351c6f9545fff54e5d9ffdf44818f21b1e8eb9cb0347b0a8745dc
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/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.43758178807986087
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3539 | LCC = 0.8074 | SRCC = 0.8132 ]
8
+ INFO:root:Average loss=0.27459769141979706
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3049 | LCC = 0.8408 | SRCC = 0.8401 ]
11
+ INFO:root:Average loss=0.2429573260820829
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3024 | LCC = 0.8515 | SRCC = 0.8493 ]
14
+ INFO:root:Average loss=0.22346308005925936
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2774 | LCC = 0.8556 | SRCC = 0.8543 ]
17
+ INFO:root:Average loss=0.21441376257019165
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2575 | LCC = 0.8587 | SRCC = 0.8564 ]
20
+ INFO:root:Average loss=0.20390632767707872
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2398 | LCC = 0.8612 | SRCC = 0.8603 ]
23
+ INFO:root:Average loss=0.1827615274068637
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2447 | LCC = 0.8647 | SRCC = 0.8631 ]
26
+ INFO:root:Average loss=0.17666180852131966
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2448 | LCC = 0.8626 | SRCC = 0.8612 ]
29
+ INFO:root:Average loss=0.17401467999204612
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2554 | LCC = 0.8610 | SRCC = 0.8603 ]
32
+ INFO:root:Average loss=0.15118789195250243
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2648 | LCC = 0.8550 | SRCC = 0.8547 ]
35
+ INFO:root:Average loss=0.14323040499137
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2619 | LCC = 0.8552 | SRCC = 0.8540 ]
38
+ INFO:root:Average loss=0.13883352661744142
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2436 | LCC = 0.8555 | SRCC = 0.8536 ]
41
+ INFO:root:Average loss=0.13759343230571502
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2495 | LCC = 0.8558 | SRCC = 0.8549 ]
44
+ INFO:root:Average loss=0.12571560648771432
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2366 | LCC = 0.8557 | SRCC = 0.8539 ]
47
+ INFO:root:Average loss=0.11170433576290424
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2476 | LCC = 0.8527 | SRCC = 0.8527 ]
50
+ INFO:root:Average loss=0.11621865797310303
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2430 | LCC = 0.8506 | SRCC = 0.8492 ]
53
+ INFO:root:Average loss=0.11055554239413677
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2520 | LCC = 0.8499 | SRCC = 0.8491 ]
56
+ INFO:root:Average loss=0.09552048122844635
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2442 | LCC = 0.8482 | SRCC = 0.8477 ]
59
+ INFO:root:Average loss=0.0940097991663676
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2473 | LCC = 0.8450 | SRCC = 0.8442 ]
62
+ INFO:root:Average loss=0.08238960525546318
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2469 | LCC = 0.8447 | SRCC = 0.8448 ]
65
+ INFO:root:Average loss=0.08315739842752616
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2684 | LCC = 0.8465 | SRCC = 0.8446 ]
68
+ INFO:root:Average loss=0.07951096015480849
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2539 | LCC = 0.8463 | SRCC = 0.8448 ]
71
+ INFO:root:Average loss=0.0808035160582035
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2467 | LCC = 0.8466 | SRCC = 0.8462 ]
74
+ INFO:root:Average loss=0.06967812523436852
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2423 | LCC = 0.8485 | SRCC = 0.8481 ]
77
+ INFO:root:Average loss=0.06693637542999707
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2659 | LCC = 0.8458 | SRCC = 0.8452 ]
80
+ INFO:root:Average loss=0.06728526051992025
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2427 | LCC = 0.8468 | SRCC = 0.8467 ]
83
+ INFO:root:Average loss=0.06428273881857212
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2512 | LCC = 0.8420 | SRCC = 0.8412 ]
86
+ INFO:root:Average loss=0.06212889455640928
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2510 | LCC = 0.8475 | SRCC = 0.8468 ]
89
+ INFO:root:Average loss=0.05527650297452242
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2555 | LCC = 0.8418 | SRCC = 0.8410 ]
92
+ INFO:root:Average loss=0.05925052374219283
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2549 | LCC = 0.8511 | SRCC = 0.8518 ]
95
+ INFO:root:Average loss=0.0530470193674167
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2690 | LCC = 0.8456 | SRCC = 0.8437 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2736 | LCC = 0.8476 | SRCC = 0.8479 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/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_2epoch/layer7/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c005cc74d5043872a925bfe06ff3108e12e70282024c0b9cb22bc4a514fe008
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e46e158aa41028fdf4ecd033cb845311dcaf0fafaf73575ad130afc911de705
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer7/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.452056314318608
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3180 | LCC = 0.8033 | SRCC = 0.8126 ]
8
+ INFO:root:Average loss=0.28075962494581175
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3133 | LCC = 0.8388 | SRCC = 0.8359 ]
11
+ INFO:root:Average loss=0.24883588747336313
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2865 | LCC = 0.8490 | SRCC = 0.8459 ]
14
+ INFO:root:Average loss=0.22087921412327352
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2560 | LCC = 0.8562 | SRCC = 0.8534 ]
17
+ INFO:root:Average loss=0.20980743165963736
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2687 | LCC = 0.8581 | SRCC = 0.8557 ]
20
+ INFO:root:Average loss=0.19847632696231207
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2448 | LCC = 0.8600 | SRCC = 0.8589 ]
23
+ INFO:root:Average loss=0.17746730502217245
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2413 | LCC = 0.8595 | SRCC = 0.8583 ]
26
+ INFO:root:Average loss=0.16760278779726762
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2404 | LCC = 0.8605 | SRCC = 0.8585 ]
29
+ INFO:root:Average loss=0.15927141522749877
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2363 | LCC = 0.8587 | SRCC = 0.8557 ]
32
+ INFO:root:Average loss=0.14369378086083975
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2501 | LCC = 0.8568 | SRCC = 0.8550 ]
35
+ INFO:root:Average loss=0.13832736101288062
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2567 | LCC = 0.8544 | SRCC = 0.8524 ]
38
+ INFO:root:Average loss=0.12984634582430887
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2514 | LCC = 0.8541 | SRCC = 0.8525 ]
41
+ INFO:root:Average loss=0.11912940853299239
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2415 | LCC = 0.8569 | SRCC = 0.8557 ]
44
+ INFO:root:Average loss=0.11701363540039612
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2581 | LCC = 0.8531 | SRCC = 0.8513 ]
47
+ INFO:root:Average loss=0.10647028144926597
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2474 | LCC = 0.8503 | SRCC = 0.8478 ]
50
+ INFO:root:Average loss=0.09998841378360222
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2427 | LCC = 0.8517 | SRCC = 0.8514 ]
53
+ INFO:root:Average loss=0.09776374286948106
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2416 | LCC = 0.8488 | SRCC = 0.8487 ]
56
+ INFO:root:Average loss=0.09849136690489757
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2510 | LCC = 0.8530 | SRCC = 0.8514 ]
59
+ INFO:root:Average loss=0.08787564326746342
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2531 | LCC = 0.8468 | SRCC = 0.8472 ]
62
+ INFO:root:Average loss=0.08047295696078202
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2503 | LCC = 0.8481 | SRCC = 0.8469 ]
65
+ INFO:root:Average loss=0.0800807113544299
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2806 | LCC = 0.8476 | SRCC = 0.8461 ]
68
+ INFO:root:Average loss=0.07766213654898681
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2480 | LCC = 0.8466 | SRCC = 0.8454 ]
71
+ INFO:root:Average loss=0.06867456698837952
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2421 | LCC = 0.8462 | SRCC = 0.8455 ]
74
+ INFO:root:Average loss=0.07454978598233981
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2873 | LCC = 0.8408 | SRCC = 0.8422 ]
77
+ INFO:root:Average loss=0.06894614456746823
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2485 | LCC = 0.8482 | SRCC = 0.8472 ]
80
+ INFO:root:Average loss=0.06405419498108901
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2435 | LCC = 0.8488 | SRCC = 0.8475 ]
83
+ INFO:root:Average loss=0.06023827629784743
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2630 | LCC = 0.8463 | SRCC = 0.8452 ]
86
+ INFO:root:Average loss=0.05772898284097513
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2517 | LCC = 0.8441 | SRCC = 0.8437 ]
89
+ INFO:root:Average loss=0.05193462480719273
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2475 | LCC = 0.8469 | SRCC = 0.8466 ]
92
+ INFO:root:Average loss=0.05206845416568029
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2494 | LCC = 0.8462 | SRCC = 0.8451 ]
95
+ INFO:root:Average loss=0.048558820803196005
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2480 | LCC = 0.8450 | SRCC = 0.8443 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2674 | LCC = 0.8480 | SRCC = 0.8486 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a75ea57d1c60fcc537a34a6c7f3689c31eaec88efc4140bafdc7bfb668461c43
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run0/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2cffb8133bf1e41da74124aac9cb28a7eb05e9d22f8f5ddda8411c7e55d06b40
3
+ size 2749638
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.47223941676127607
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3751 | LCC = 0.7904 | SRCC = 0.8011 ]
8
+ INFO:root:Average loss=0.27982731278126055
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3302 | LCC = 0.8303 | SRCC = 0.8308 ]
11
+ INFO:root:Average loss=0.2549691217449995
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2838 | LCC = 0.8463 | SRCC = 0.8453 ]
14
+ INFO:root:Average loss=0.23500471390210664
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2539 | LCC = 0.8503 | SRCC = 0.8517 ]
17
+ INFO:root:Average loss=0.21987843398864454
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2718 | LCC = 0.8506 | SRCC = 0.8522 ]
20
+ INFO:root:Average loss=0.2170359866741376
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2457 | LCC = 0.8586 | SRCC = 0.8584 ]
23
+ INFO:root:Average loss=0.20060518842477065
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2286 | LCC = 0.8626 | SRCC = 0.8630 ]
26
+ INFO:root:Average loss=0.19284183226334742
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2629 | LCC = 0.8586 | SRCC = 0.8582 ]
29
+ INFO:root:Average loss=0.17678190242403594
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2560 | LCC = 0.8595 | SRCC = 0.8614 ]
32
+ INFO:root:Average loss=0.17932205007244378
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2733 | LCC = 0.8643 | SRCC = 0.8641 ]
35
+ INFO:root:Average loss=0.16425106836817202
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2289 | LCC = 0.8613 | SRCC = 0.8604 ]
38
+ INFO:root:Average loss=0.16195728620275474
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2871 | LCC = 0.8548 | SRCC = 0.8567 ]
41
+ INFO:root:Average loss=0.14901328545350295
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2531 | LCC = 0.8595 | SRCC = 0.8583 ]
44
+ INFO:root:Average loss=0.1450820328333439
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2588 | LCC = 0.8582 | SRCC = 0.8622 ]
47
+ INFO:root:Average loss=0.14872629548876715
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2499 | LCC = 0.8616 | SRCC = 0.8629 ]
50
+ INFO:root:Average loss=0.13497571045389542
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2722 | LCC = 0.8587 | SRCC = 0.8610 ]
53
+ INFO:root:Average loss=0.12823205307508126
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2513 | LCC = 0.8575 | SRCC = 0.8599 ]
56
+ INFO:root:Average loss=0.11853662391121571
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2392 | LCC = 0.8583 | SRCC = 0.8596 ]
59
+ INFO:root:Average loss=0.1215384110617332
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2400 | LCC = 0.8624 | SRCC = 0.8627 ]
62
+ INFO:root:Average loss=0.11138051327986595
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2692 | LCC = 0.8510 | SRCC = 0.8559 ]
65
+ INFO:root:Average loss=0.10720421330860028
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2219 | LCC = 0.8603 | SRCC = 0.8609 ]
68
+ INFO:root:Average loss=0.10368176158039998
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2606 | LCC = 0.8563 | SRCC = 0.8613 ]
71
+ INFO:root:Average loss=0.09395950932342273
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2329 | LCC = 0.8539 | SRCC = 0.8549 ]
74
+ INFO:root:Average loss=0.0912096606901823
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2483 | LCC = 0.8518 | SRCC = 0.8572 ]
77
+ INFO:root:Average loss=0.0967549738020469
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2274 | LCC = 0.8592 | SRCC = 0.8603 ]
80
+ INFO:root:Average loss=0.08598366322425696
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2381 | LCC = 0.8554 | SRCC = 0.8547 ]
83
+ INFO:root:Average loss=0.08496240774790446
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2299 | LCC = 0.8567 | SRCC = 0.8560 ]
86
+ INFO:root:Average loss=0.07885864548958264
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2292 | LCC = 0.8565 | SRCC = 0.8591 ]
89
+ INFO:root:Average loss=0.07814654598060326
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2370 | LCC = 0.8523 | SRCC = 0.8532 ]
92
+ INFO:root:Average loss=0.0762549732835629
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2385 | LCC = 0.8525 | SRCC = 0.8537 ]
95
+ INFO:root:Average loss=0.07473513718025807
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2510 | LCC = 0.8476 | SRCC = 0.8498 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2667 | LCC = 0.8646 | SRCC = 0.8623 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run1/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdafd90b7bc12c729c4df2e3a1609d1346e0c4c2e5ebf3bd6b18b0c23d7c7de3
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run1/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dd2f52000cdb89be8a5b10597c15b96ad3e0745365a10c38fbe2eb634132fef
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.41303758246776384
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3732 | LCC = 0.8069 | SRCC = 0.8122 ]
8
+ INFO:root:Average loss=0.285670280838624
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3252 | LCC = 0.8318 | SRCC = 0.8336 ]
11
+ INFO:root:Average loss=0.2465732741432312
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2782 | LCC = 0.8431 | SRCC = 0.8426 ]
14
+ INFO:root:Average loss=0.23031453559031853
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2695 | LCC = 0.8469 | SRCC = 0.8471 ]
17
+ INFO:root:Average loss=0.2165564417074888
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2533 | LCC = 0.8569 | SRCC = 0.8579 ]
20
+ INFO:root:Average loss=0.20914513503129667
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2484 | LCC = 0.8551 | SRCC = 0.8553 ]
23
+ INFO:root:Average loss=0.19378782894748908
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2830 | LCC = 0.8554 | SRCC = 0.8556 ]
26
+ INFO:root:Average loss=0.19045280875303808
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2577 | LCC = 0.8613 | SRCC = 0.8614 ]
29
+ INFO:root:Average loss=0.1825267035418596
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2559 | LCC = 0.8612 | SRCC = 0.8587 ]
32
+ INFO:root:Average loss=0.1781048531142565
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2480 | LCC = 0.8628 | SRCC = 0.8614 ]
35
+ INFO:root:Average loss=0.1567079455424578
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2610 | LCC = 0.8596 | SRCC = 0.8582 ]
38
+ INFO:root:Average loss=0.15464371643387353
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2663 | LCC = 0.8602 | SRCC = 0.8593 ]
41
+ INFO:root:Average loss=0.15943133668639722
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2434 | LCC = 0.8588 | SRCC = 0.8588 ]
44
+ INFO:root:Average loss=0.13974300399422646
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2527 | LCC = 0.8597 | SRCC = 0.8590 ]
47
+ INFO:root:Average loss=0.13362917944024771
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2506 | LCC = 0.8576 | SRCC = 0.8567 ]
50
+ INFO:root:Average loss=0.14108931206357786
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2575 | LCC = 0.8563 | SRCC = 0.8579 ]
53
+ INFO:root:Average loss=0.12080439170583701
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2339 | LCC = 0.8572 | SRCC = 0.8572 ]
56
+ INFO:root:Average loss=0.12141439557457581
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2522 | LCC = 0.8558 | SRCC = 0.8563 ]
59
+ INFO:root:Average loss=0.11084135382985458
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2263 | LCC = 0.8595 | SRCC = 0.8601 ]
62
+ INFO:root:Average loss=0.10768489721111763
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2399 | LCC = 0.8555 | SRCC = 0.8557 ]
65
+ INFO:root:Average loss=0.10416071470349263
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2568 | LCC = 0.8535 | SRCC = 0.8535 ]
68
+ INFO:root:Average loss=0.1003027319526061
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2479 | LCC = 0.8515 | SRCC = 0.8523 ]
71
+ INFO:root:Average loss=0.09413785067124245
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2455 | LCC = 0.8547 | SRCC = 0.8538 ]
74
+ INFO:root:Average loss=0.09474462060592113
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2563 | LCC = 0.8475 | SRCC = 0.8485 ]
77
+ INFO:root:Average loss=0.08715413195582536
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2635 | LCC = 0.8497 | SRCC = 0.8499 ]
80
+ INFO:root:Average loss=0.08358084114316182
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2458 | LCC = 0.8492 | SRCC = 0.8482 ]
83
+ INFO:root:Average loss=0.08539289790086257
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2477 | LCC = 0.8525 | SRCC = 0.8543 ]
86
+ INFO:root:Average loss=0.07988832332193851
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2406 | LCC = 0.8528 | SRCC = 0.8537 ]
89
+ INFO:root:Average loss=0.0792451973240345
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2300 | LCC = 0.8561 | SRCC = 0.8560 ]
92
+ INFO:root:Average loss=0.07189440622161596
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2456 | LCC = 0.8487 | SRCC = 0.8490 ]
95
+ INFO:root:Average loss=0.07597707555844234
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2339 | LCC = 0.8544 | SRCC = 0.8541 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2507 | LCC = 0.8575 | SRCC = 0.8566 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:28554270ea3746945a70254d3c16fa1891e297fe95b35782b784777732203d60
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab01059d7aa73d848a850fac307178a448ea569cade2e633670f4f4112b46571
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.4284714019069305
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.4288 | LCC = 0.8063 | SRCC = 0.8101 ]
8
+ INFO:root:Average loss=0.2851868722683344
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3220 | LCC = 0.8339 | SRCC = 0.8350 ]
11
+ INFO:root:Average loss=0.24797301987806955
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3159 | LCC = 0.8416 | SRCC = 0.8408 ]
14
+ INFO:root:Average loss=0.23648603585286018
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.3584 | LCC = 0.8436 | SRCC = 0.8444 ]
17
+ INFO:root:Average loss=0.21922305016181406
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.3090 | LCC = 0.8522 | SRCC = 0.8512 ]
20
+ INFO:root:Average loss=0.23053583999474844
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2908 | LCC = 0.8516 | SRCC = 0.8494 ]
23
+ INFO:root:Average loss=0.20694475363080317
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2505 | LCC = 0.8596 | SRCC = 0.8567 ]
26
+ INFO:root:Average loss=0.1895792339092646
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2652 | LCC = 0.8616 | SRCC = 0.8609 ]
29
+ INFO:root:Average loss=0.17921728335129908
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2995 | LCC = 0.8570 | SRCC = 0.8550 ]
32
+ INFO:root:Average loss=0.16879250319340292
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2540 | LCC = 0.8542 | SRCC = 0.8517 ]
35
+ INFO:root:Average loss=0.16210990771651268
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2948 | LCC = 0.8568 | SRCC = 0.8579 ]
38
+ INFO:root:Average loss=0.15975914952846673
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.3007 | LCC = 0.8533 | SRCC = 0.8534 ]
41
+ INFO:root:Average loss=0.1476324700201169
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2552 | LCC = 0.8567 | SRCC = 0.8565 ]
44
+ INFO:root:Average loss=0.13965487948212868
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2529 | LCC = 0.8545 | SRCC = 0.8533 ]
47
+ INFO:root:Average loss=0.1314195753672184
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2787 | LCC = 0.8546 | SRCC = 0.8535 ]
50
+ INFO:root:Average loss=0.12969911471009254
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2740 | LCC = 0.8548 | SRCC = 0.8525 ]
53
+ INFO:root:Average loss=0.12821087948022744
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2554 | LCC = 0.8507 | SRCC = 0.8486 ]
56
+ INFO:root:Average loss=0.12309678414693245
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2384 | LCC = 0.8510 | SRCC = 0.8487 ]
59
+ INFO:root:Average loss=0.10895248215932113
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.3008 | LCC = 0.8464 | SRCC = 0.8450 ]
62
+ INFO:root:Average loss=0.10654697099175209
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2488 | LCC = 0.8468 | SRCC = 0.8454 ]
65
+ INFO:root:Average loss=0.1012338790564965
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.3068 | LCC = 0.8422 | SRCC = 0.8417 ]
68
+ INFO:root:Average loss=0.10265656675283726
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2414 | LCC = 0.8506 | SRCC = 0.8495 ]
71
+ INFO:root:Average loss=0.0958816347978054
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2623 | LCC = 0.8450 | SRCC = 0.8436 ]
74
+ INFO:root:Average loss=0.08844794638645954
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2644 | LCC = 0.8445 | SRCC = 0.8442 ]
77
+ INFO:root:Average loss=0.08726239767976296
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2722 | LCC = 0.8456 | SRCC = 0.8444 ]
80
+ INFO:root:Average loss=0.08903845958411694
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.3081 | LCC = 0.8481 | SRCC = 0.8483 ]
83
+ INFO:root:Average loss=0.08786079831994496
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2508 | LCC = 0.8497 | SRCC = 0.8485 ]
86
+ INFO:root:Average loss=0.08469467180279586
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2450 | LCC = 0.8485 | SRCC = 0.8476 ]
89
+ INFO:root:Average loss=0.07920091355649325
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2528 | LCC = 0.8470 | SRCC = 0.8465 ]
92
+ INFO:root:Average loss=0.07507136316062549
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2453 | LCC = 0.8480 | SRCC = 0.8474 ]
95
+ INFO:root:Average loss=0.0728788711130619
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2559 | LCC = 0.8486 | SRCC = 0.8482 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2579 | LCC = 0.8554 | SRCC = 0.8532 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39e04901f0c7684dec0f71c23996942b5de1561e7b298e98635d26038f918778
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:817ef44e113973ca2620964878941835dd0cddb7070ab69523e6da0414097fc5
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.40541908622552186
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3292 | LCC = 0.8130 | SRCC = 0.8154 ]
8
+ INFO:root:Average loss=0.27299155103854644
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.2932 | LCC = 0.8382 | SRCC = 0.8358 ]
11
+ INFO:root:Average loss=0.2485059251387914
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2642 | LCC = 0.8462 | SRCC = 0.8457 ]
14
+ INFO:root:Average loss=0.2358058819021934
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2710 | LCC = 0.8525 | SRCC = 0.8533 ]
17
+ INFO:root:Average loss=0.2178884758017002
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2632 | LCC = 0.8556 | SRCC = 0.8525 ]
20
+ INFO:root:Average loss=0.21346342353484568
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2489 | LCC = 0.8528 | SRCC = 0.8503 ]
23
+ INFO:root:Average loss=0.20036793786745805
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2706 | LCC = 0.8604 | SRCC = 0.8579 ]
26
+ INFO:root:Average loss=0.18231598717662004
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2438 | LCC = 0.8625 | SRCC = 0.8604 ]
29
+ INFO:root:Average loss=0.17464523705152366
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2510 | LCC = 0.8567 | SRCC = 0.8542 ]
32
+ INFO:root:Average loss=0.16686499223877221
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2490 | LCC = 0.8642 | SRCC = 0.8619 ]
35
+ INFO:root:Average loss=0.15523532529671988
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2472 | LCC = 0.8595 | SRCC = 0.8564 ]
38
+ INFO:root:Average loss=0.14768393203998223
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2777 | LCC = 0.8592 | SRCC = 0.8579 ]
41
+ INFO:root:Average loss=0.14827872946476325
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2755 | LCC = 0.8584 | SRCC = 0.8579 ]
44
+ INFO:root:Average loss=0.1361596627304187
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2317 | LCC = 0.8554 | SRCC = 0.8531 ]
47
+ INFO:root:Average loss=0.12682292954279825
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2496 | LCC = 0.8557 | SRCC = 0.8531 ]
50
+ INFO:root:Average loss=0.12293366266366763
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2605 | LCC = 0.8584 | SRCC = 0.8576 ]
53
+ INFO:root:Average loss=0.11543400318194659
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2605 | LCC = 0.8524 | SRCC = 0.8505 ]
56
+ INFO:root:Average loss=0.11308515215149292
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2459 | LCC = 0.8544 | SRCC = 0.8535 ]
59
+ INFO:root:Average loss=0.11106220088326015
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2462 | LCC = 0.8513 | SRCC = 0.8499 ]
62
+ INFO:root:Average loss=0.10959643316574585
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2388 | LCC = 0.8552 | SRCC = 0.8537 ]
65
+ INFO:root:Average loss=0.10280151264025615
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2367 | LCC = 0.8572 | SRCC = 0.8561 ]
68
+ INFO:root:Average loss=0.09583375765345035
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2472 | LCC = 0.8525 | SRCC = 0.8525 ]
71
+ INFO:root:Average loss=0.09108715327695394
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2474 | LCC = 0.8538 | SRCC = 0.8519 ]
74
+ INFO:root:Average loss=0.08602023435135682
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2464 | LCC = 0.8517 | SRCC = 0.8498 ]
77
+ INFO:root:Average loss=0.08819783736880009
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2689 | LCC = 0.8464 | SRCC = 0.8452 ]
80
+ INFO:root:Average loss=0.0810032367037657
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2375 | LCC = 0.8509 | SRCC = 0.8496 ]
83
+ INFO:root:Average loss=0.07612811587750912
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2506 | LCC = 0.8500 | SRCC = 0.8486 ]
86
+ INFO:root:Average loss=0.08309972878450002
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2633 | LCC = 0.8494 | SRCC = 0.8484 ]
89
+ INFO:root:Average loss=0.07546157108094448
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2501 | LCC = 0.8498 | SRCC = 0.8478 ]
92
+ INFO:root:Average loss=0.0741928331553936
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2582 | LCC = 0.8511 | SRCC = 0.8502 ]
95
+ INFO:root:Average loss=0.06476368224964692
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2410 | LCC = 0.8492 | SRCC = 0.8491 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2692 | LCC = 0.8569 | SRCC = 0.8593 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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_2epoch/layer8/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a7bbb8e0472ccf1fbd917353cb8eda0f47b45934cd5cfc6aeee445c8fe214b0
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b0f3f4e414f8b12d9d5dcda1231a614be423c83892f5772f68e636a1dd89c78
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer8/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.43741939942806196
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3538 | LCC = 0.7955 | SRCC = 0.7995 ]
8
+ INFO:root:Average loss=0.2892684185734162
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3225 | LCC = 0.8224 | SRCC = 0.8260 ]
11
+ INFO:root:Average loss=0.24687125247258407
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.2957 | LCC = 0.8453 | SRCC = 0.8444 ]
14
+ INFO:root:Average loss=0.22453046322633058
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2956 | LCC = 0.8495 | SRCC = 0.8497 ]
17
+ INFO:root:Average loss=0.2161407819352089
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.2601 | LCC = 0.8594 | SRCC = 0.8583 ]
20
+ INFO:root:Average loss=0.2075838952874526
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2637 | LCC = 0.8630 | SRCC = 0.8620 ]
23
+ INFO:root:Average loss=0.2106938322003071
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2440 | LCC = 0.8615 | SRCC = 0.8610 ]
26
+ INFO:root:Average loss=0.1833732172082632
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2426 | LCC = 0.8628 | SRCC = 0.8626 ]
29
+ INFO:root:Average loss=0.17070448857087356
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2519 | LCC = 0.8630 | SRCC = 0.8629 ]
32
+ INFO:root:Average loss=0.16309963032985345
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2384 | LCC = 0.8641 | SRCC = 0.8645 ]
35
+ INFO:root:Average loss=0.15788956072468024
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2264 | LCC = 0.8653 | SRCC = 0.8652 ]
38
+ INFO:root:Average loss=0.15759127290967184
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.3060 | LCC = 0.8598 | SRCC = 0.8622 ]
41
+ INFO:root:Average loss=0.15851521339171973
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2240 | LCC = 0.8663 | SRCC = 0.8675 ]
44
+ INFO:root:Average loss=0.1394150136755063
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2274 | LCC = 0.8664 | SRCC = 0.8670 ]
47
+ INFO:root:Average loss=0.13565322651694983
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2362 | LCC = 0.8644 | SRCC = 0.8649 ]
50
+ INFO:root:Average loss=0.12542991130015788
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2336 | LCC = 0.8641 | SRCC = 0.8647 ]
53
+ INFO:root:Average loss=0.11608249178299537
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2440 | LCC = 0.8606 | SRCC = 0.8614 ]
56
+ INFO:root:Average loss=0.11595186696220668
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2413 | LCC = 0.8592 | SRCC = 0.8591 ]
59
+ INFO:root:Average loss=0.11024917996464631
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2382 | LCC = 0.8596 | SRCC = 0.8605 ]
62
+ INFO:root:Average loss=0.10416477918624878
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2674 | LCC = 0.8542 | SRCC = 0.8535 ]
65
+ INFO:root:Average loss=0.09718882345045224
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2379 | LCC = 0.8586 | SRCC = 0.8599 ]
68
+ INFO:root:Average loss=0.09196937079421985
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2561 | LCC = 0.8534 | SRCC = 0.8541 ]
71
+ INFO:root:Average loss=0.09043105338246395
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2331 | LCC = 0.8574 | SRCC = 0.8587 ]
74
+ INFO:root:Average loss=0.08995892900304917
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2339 | LCC = 0.8598 | SRCC = 0.8597 ]
77
+ INFO:root:Average loss=0.08405351543273681
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2510 | LCC = 0.8571 | SRCC = 0.8573 ]
80
+ INFO:root:Average loss=0.07768201933075221
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2522 | LCC = 0.8562 | SRCC = 0.8572 ]
83
+ INFO:root:Average loss=0.0780353932044445
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2251 | LCC = 0.8571 | SRCC = 0.8581 ]
86
+ INFO:root:Average loss=0.07864247553814681
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2314 | LCC = 0.8559 | SRCC = 0.8565 ]
89
+ INFO:root:Average loss=0.07323050661346851
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2243 | LCC = 0.8572 | SRCC = 0.8585 ]
92
+ INFO:root:Average loss=0.07265138511474316
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2550 | LCC = 0.8526 | SRCC = 0.8534 ]
95
+ INFO:root:Average loss=0.0646772900930582
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2387 | LCC = 0.8538 | SRCC = 0.8545 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2626 | LCC = 0.8533 | SRCC = 0.8526 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run0/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8712c99bb3cbfe5a0a5e91f3d2054ea82f6084c0e6ed874a46179e4a3bf6dca8
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run0/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87bc07c53d1a2118dfc2a1a194b372b56b536e961461d15da3d0c288e1571401
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.40266313346532673
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3359 | LCC = 0.7966 | SRCC = 0.7982 ]
8
+ INFO:root:Average loss=0.29929162313540775
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3918 | LCC = 0.8220 | SRCC = 0.8244 ]
11
+ INFO:root:Average loss=0.2818653656122012
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3207 | LCC = 0.8372 | SRCC = 0.8360 ]
14
+ INFO:root:Average loss=0.263999010125796
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.2658 | LCC = 0.8363 | SRCC = 0.8379 ]
17
+ INFO:root:Average loss=0.2569745114216438
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.3762 | LCC = 0.8436 | SRCC = 0.8459 ]
20
+ INFO:root:Average loss=0.24703097190612403
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2841 | LCC = 0.8474 | SRCC = 0.8477 ]
23
+ INFO:root:Average loss=0.2624498519759912
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.3735 | LCC = 0.8507 | SRCC = 0.8502 ]
26
+ INFO:root:Average loss=0.23869414455615556
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2764 | LCC = 0.8506 | SRCC = 0.8507 ]
29
+ INFO:root:Average loss=0.21669101361662912
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2513 | LCC = 0.8581 | SRCC = 0.8578 ]
32
+ INFO:root:Average loss=0.20718366289750123
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.3034 | LCC = 0.8592 | SRCC = 0.8612 ]
35
+ INFO:root:Average loss=0.2161941912311774
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.3129 | LCC = 0.8583 | SRCC = 0.8575 ]
38
+ INFO:root:Average loss=0.21269655743470559
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.3040 | LCC = 0.8601 | SRCC = 0.8583 ]
41
+ INFO:root:Average loss=0.20141950383400306
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2737 | LCC = 0.8605 | SRCC = 0.8580 ]
44
+ INFO:root:Average loss=0.20354594595921346
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2812 | LCC = 0.8585 | SRCC = 0.8570 ]
47
+ INFO:root:Average loss=0.19188268568653327
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.3086 | LCC = 0.8573 | SRCC = 0.8610 ]
50
+ INFO:root:Average loss=0.1888957713276912
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.3459 | LCC = 0.8562 | SRCC = 0.8609 ]
53
+ INFO:root:Average loss=0.18730529254445663
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2625 | LCC = 0.8589 | SRCC = 0.8570 ]
56
+ INFO:root:Average loss=0.1779670580648459
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2522 | LCC = 0.8563 | SRCC = 0.8567 ]
59
+ INFO:root:Average loss=0.1712119739789229
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2592 | LCC = 0.8588 | SRCC = 0.8585 ]
62
+ INFO:root:Average loss=0.17276713022818932
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2630 | LCC = 0.8600 | SRCC = 0.8582 ]
65
+ INFO:root:Average loss=0.1593610692100647
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2969 | LCC = 0.8575 | SRCC = 0.8592 ]
68
+ INFO:root:Average loss=0.1659397159058314
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2753 | LCC = 0.8618 | SRCC = 0.8607 ]
71
+ INFO:root:Average loss=0.1549930579196184
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2566 | LCC = 0.8619 | SRCC = 0.8616 ]
74
+ INFO:root:Average loss=0.16745337662406456
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2867 | LCC = 0.8570 | SRCC = 0.8570 ]
77
+ INFO:root:Average loss=0.15673001817403695
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.3009 | LCC = 0.8588 | SRCC = 0.8578 ]
80
+ INFO:root:Average loss=0.14925719224489653
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2394 | LCC = 0.8603 | SRCC = 0.8587 ]
83
+ INFO:root:Average loss=0.15450562527164435
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2645 | LCC = 0.8614 | SRCC = 0.8594 ]
86
+ INFO:root:Average loss=0.1366595337406183
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2373 | LCC = 0.8575 | SRCC = 0.8565 ]
89
+ INFO:root:Average loss=0.13738801101079354
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2639 | LCC = 0.8533 | SRCC = 0.8522 ]
92
+ INFO:root:Average loss=0.1267202989413188
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2985 | LCC = 0.8510 | SRCC = 0.8510 ]
95
+ INFO:root:Average loss=0.12687058336077592
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2258 | LCC = 0.8599 | SRCC = 0.8586 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2939 | LCC = 0.8493 | SRCC = 0.8522 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run1/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c236fa89aa193ebad1e250f365e8baee945fb63a83c1f20b9871743db6e32abb
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run1/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f35b472145f937f13071729146a86316d7127143bfacc1e8e7a56a24b259c011
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.4648682929766484
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3894 | LCC = 0.7893 | SRCC = 0.7919 ]
8
+ INFO:root:Average loss=0.29751755621952886
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3217 | LCC = 0.8200 | SRCC = 0.8215 ]
11
+ INFO:root:Average loss=0.2782701930174461
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3154 | LCC = 0.8149 | SRCC = 0.8217 ]
14
+ INFO:root:Average loss=0.2735561869847469
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.3193 | LCC = 0.8401 | SRCC = 0.8377 ]
17
+ INFO:root:Average loss=0.26198548556138307
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.3044 | LCC = 0.8434 | SRCC = 0.8425 ]
20
+ INFO:root:Average loss=0.24024102378350037
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2920 | LCC = 0.8517 | SRCC = 0.8515 ]
23
+ INFO:root:Average loss=0.2412498067968931
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2637 | LCC = 0.8512 | SRCC = 0.8502 ]
26
+ INFO:root:Average loss=0.24265832874255303
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.3037 | LCC = 0.8526 | SRCC = 0.8512 ]
29
+ INFO:root:Average loss=0.2187001244762005
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2693 | LCC = 0.8561 | SRCC = 0.8561 ]
32
+ INFO:root:Average loss=0.2090743953982989
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2790 | LCC = 0.8521 | SRCC = 0.8535 ]
35
+ INFO:root:Average loss=0.20778891501518396
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2772 | LCC = 0.8602 | SRCC = 0.8603 ]
38
+ INFO:root:Average loss=0.21923466695424837
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2922 | LCC = 0.8600 | SRCC = 0.8575 ]
41
+ INFO:root:Average loss=0.19139932201076776
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.3009 | LCC = 0.8639 | SRCC = 0.8625 ]
44
+ INFO:root:Average loss=0.19710553341951126
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.3082 | LCC = 0.8659 | SRCC = 0.8639 ]
47
+ INFO:root:Average loss=0.18678924231193003
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2350 | LCC = 0.8624 | SRCC = 0.8617 ]
50
+ INFO:root:Average loss=0.18032619930230653
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2665 | LCC = 0.8625 | SRCC = 0.8607 ]
53
+ INFO:root:Average loss=0.1813369193711342
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2409 | LCC = 0.8636 | SRCC = 0.8624 ]
56
+ INFO:root:Average loss=0.16888423808492148
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2555 | LCC = 0.8607 | SRCC = 0.8598 ]
59
+ INFO:root:Average loss=0.17025642154308465
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2545 | LCC = 0.8646 | SRCC = 0.8632 ]
62
+ INFO:root:Average loss=0.16329536577447867
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2634 | LCC = 0.8619 | SRCC = 0.8599 ]
65
+ INFO:root:Average loss=0.16247353483086976
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2443 | LCC = 0.8630 | SRCC = 0.8624 ]
68
+ INFO:root:Average loss=0.160555169177361
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2394 | LCC = 0.8619 | SRCC = 0.8615 ]
71
+ INFO:root:Average loss=0.163032844185065
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2617 | LCC = 0.8571 | SRCC = 0.8571 ]
74
+ INFO:root:Average loss=0.15858055183138603
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2323 | LCC = 0.8579 | SRCC = 0.8590 ]
77
+ INFO:root:Average loss=0.14591626756084272
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2573 | LCC = 0.8587 | SRCC = 0.8566 ]
80
+ INFO:root:Average loss=0.14349040169364366
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2601 | LCC = 0.8614 | SRCC = 0.8605 ]
83
+ INFO:root:Average loss=0.13605844850341478
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2414 | LCC = 0.8572 | SRCC = 0.8570 ]
86
+ INFO:root:Average loss=0.1352860598036876
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2451 | LCC = 0.8560 | SRCC = 0.8558 ]
89
+ INFO:root:Average loss=0.13141561834475932
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2441 | LCC = 0.8581 | SRCC = 0.8571 ]
92
+ INFO:root:Average loss=0.1243367326947359
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2485 | LCC = 0.8564 | SRCC = 0.8564 ]
95
+ INFO:root:Average loss=0.12883145486315092
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2562 | LCC = 0.8565 | SRCC = 0.8565 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.3199 | LCC = 0.8519 | SRCC = 0.8508 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run2/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca08dd9b5e33413b8dfb4bfeb23a7d2e2aeab58e3e5e42db51e17a3a56114e08
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run2/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37a7c12a4e14f45e409ca9e3dea3ed4b67e48065b2e575c091175d872c1969bc
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.4380430716734666
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3651 | LCC = 0.7911 | SRCC = 0.7934 ]
8
+ INFO:root:Average loss=0.3175906286789821
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3453 | LCC = 0.8137 | SRCC = 0.8168 ]
11
+ INFO:root:Average loss=0.30156727077869266
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3076 | LCC = 0.8227 | SRCC = 0.8246 ]
14
+ INFO:root:Average loss=0.2668433584845983
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.3279 | LCC = 0.8348 | SRCC = 0.8328 ]
17
+ INFO:root:Average loss=0.2522684801847507
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.3280 | LCC = 0.8371 | SRCC = 0.8353 ]
20
+ INFO:root:Average loss=0.2482475391947306
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.3321 | LCC = 0.8457 | SRCC = 0.8466 ]
23
+ INFO:root:Average loss=0.2267345585502111
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2629 | LCC = 0.8489 | SRCC = 0.8479 ]
26
+ INFO:root:Average loss=0.23283594827621412
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.2862 | LCC = 0.8523 | SRCC = 0.8527 ]
29
+ INFO:root:Average loss=0.23233816027641296
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.3290 | LCC = 0.8566 | SRCC = 0.8573 ]
32
+ INFO:root:Average loss=0.217007499283705
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2853 | LCC = 0.8593 | SRCC = 0.8587 ]
35
+ INFO:root:Average loss=0.21657805335827363
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2816 | LCC = 0.8598 | SRCC = 0.8581 ]
38
+ INFO:root:Average loss=0.21848731402021188
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2686 | LCC = 0.8609 | SRCC = 0.8591 ]
41
+ INFO:root:Average loss=0.2011777832148931
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.3169 | LCC = 0.8572 | SRCC = 0.8586 ]
44
+ INFO:root:Average loss=0.18928339083989462
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2622 | LCC = 0.8605 | SRCC = 0.8581 ]
47
+ INFO:root:Average loss=0.19051613677770662
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2824 | LCC = 0.8590 | SRCC = 0.8566 ]
50
+ INFO:root:Average loss=0.18298456187431628
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2672 | LCC = 0.8652 | SRCC = 0.8638 ]
53
+ INFO:root:Average loss=0.17545409223590142
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2493 | LCC = 0.8627 | SRCC = 0.8610 ]
56
+ INFO:root:Average loss=0.1711289882659912
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2479 | LCC = 0.8610 | SRCC = 0.8615 ]
59
+ INFO:root:Average loss=0.1718349771048778
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2877 | LCC = 0.8571 | SRCC = 0.8563 ]
62
+ INFO:root:Average loss=0.16342714390693566
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2460 | LCC = 0.8560 | SRCC = 0.8543 ]
65
+ INFO:root:Average loss=0.17224891751240462
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2750 | LCC = 0.8583 | SRCC = 0.8568 ]
68
+ INFO:root:Average loss=0.1667442812751501
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2911 | LCC = 0.8527 | SRCC = 0.8518 ]
71
+ INFO:root:Average loss=0.15686444708934197
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2725 | LCC = 0.8543 | SRCC = 0.8517 ]
74
+ INFO:root:Average loss=0.14840580427493805
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2502 | LCC = 0.8616 | SRCC = 0.8595 ]
77
+ INFO:root:Average loss=0.15023532681740248
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2550 | LCC = 0.8574 | SRCC = 0.8574 ]
80
+ INFO:root:Average loss=0.13739874433630553
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2442 | LCC = 0.8603 | SRCC = 0.8593 ]
83
+ INFO:root:Average loss=0.13486138492440566
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2634 | LCC = 0.8524 | SRCC = 0.8510 ]
86
+ INFO:root:Average loss=0.13031689574321112
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2684 | LCC = 0.8541 | SRCC = 0.8516 ]
89
+ INFO:root:Average loss=0.13020408506958914
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2748 | LCC = 0.8530 | SRCC = 0.8532 ]
92
+ INFO:root:Average loss=0.13747126618639016
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2592 | LCC = 0.8526 | SRCC = 0.8527 ]
95
+ INFO:root:Average loss=0.12138828176718491
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2682 | LCC = 0.8466 | SRCC = 0.8427 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2706 | LCC = 0.8554 | SRCC = 0.8557 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run3/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b548a25e71148bd5546559aae6dd1ece4414ac7f19ff6949e0c017e983bbd436
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run3/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61376ce0e5e7483513dca887e6ef4f2198e93a888ed19f707b1657d6d4aae2f2
3
+ size 2749702
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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.5438558818438114
6
+ INFO:root:
7
+ [Valid][0][UTT][ MSE = 0.3763 | LCC = 0.7776 | SRCC = 0.7818 ]
8
+ INFO:root:Average loss=0.31684504716824263
9
+ INFO:root:
10
+ [Valid][1][UTT][ MSE = 0.3293 | LCC = 0.8029 | SRCC = 0.8075 ]
11
+ INFO:root:Average loss=0.29425239715820706
12
+ INFO:root:
13
+ [Valid][2][UTT][ MSE = 0.3285 | LCC = 0.8216 | SRCC = 0.8232 ]
14
+ INFO:root:Average loss=0.2700358713284517
15
+ INFO:root:
16
+ [Valid][3][UTT][ MSE = 0.3121 | LCC = 0.8352 | SRCC = 0.8358 ]
17
+ INFO:root:Average loss=0.26226788205214036
18
+ INFO:root:
19
+ [Valid][4][UTT][ MSE = 0.3116 | LCC = 0.8435 | SRCC = 0.8439 ]
20
+ INFO:root:Average loss=0.25144613820772904
21
+ INFO:root:
22
+ [Valid][5][UTT][ MSE = 0.2722 | LCC = 0.8406 | SRCC = 0.8421 ]
23
+ INFO:root:Average loss=0.24430168592012846
24
+ INFO:root:
25
+ [Valid][6][UTT][ MSE = 0.2788 | LCC = 0.8474 | SRCC = 0.8465 ]
26
+ INFO:root:Average loss=0.23114579151838255
27
+ INFO:root:
28
+ [Valid][7][UTT][ MSE = 0.3113 | LCC = 0.8413 | SRCC = 0.8448 ]
29
+ INFO:root:Average loss=0.22467450759349725
30
+ INFO:root:
31
+ [Valid][8][UTT][ MSE = 0.2778 | LCC = 0.8434 | SRCC = 0.8444 ]
32
+ INFO:root:Average loss=0.23356003715441778
33
+ INFO:root:
34
+ [Valid][9][UTT][ MSE = 0.2907 | LCC = 0.8512 | SRCC = 0.8503 ]
35
+ INFO:root:Average loss=0.21789864087716126
36
+ INFO:root:
37
+ [Valid][10][UTT][ MSE = 0.2630 | LCC = 0.8572 | SRCC = 0.8562 ]
38
+ INFO:root:Average loss=0.2068679734873466
39
+ INFO:root:
40
+ [Valid][11][UTT][ MSE = 0.2605 | LCC = 0.8578 | SRCC = 0.8578 ]
41
+ INFO:root:Average loss=0.2001887424251972
42
+ INFO:root:
43
+ [Valid][12][UTT][ MSE = 0.2562 | LCC = 0.8586 | SRCC = 0.8572 ]
44
+ INFO:root:Average loss=0.1998200210241171
45
+ INFO:root:
46
+ [Valid][13][UTT][ MSE = 0.2862 | LCC = 0.8608 | SRCC = 0.8604 ]
47
+ INFO:root:Average loss=0.1897058802155348
48
+ INFO:root:
49
+ [Valid][14][UTT][ MSE = 0.2510 | LCC = 0.8601 | SRCC = 0.8589 ]
50
+ INFO:root:Average loss=0.18851592544561777
51
+ INFO:root:
52
+ [Valid][15][UTT][ MSE = 0.2587 | LCC = 0.8573 | SRCC = 0.8566 ]
53
+ INFO:root:Average loss=0.1853850024441878
54
+ INFO:root:
55
+ [Valid][16][UTT][ MSE = 0.2748 | LCC = 0.8663 | SRCC = 0.8648 ]
56
+ INFO:root:Average loss=0.17419022197524706
57
+ INFO:root:
58
+ [Valid][17][UTT][ MSE = 0.2541 | LCC = 0.8611 | SRCC = 0.8600 ]
59
+ INFO:root:Average loss=0.17481951577923235
60
+ INFO:root:
61
+ [Valid][18][UTT][ MSE = 0.2906 | LCC = 0.8537 | SRCC = 0.8595 ]
62
+ INFO:root:Average loss=0.16672851966741759
63
+ INFO:root:
64
+ [Valid][19][UTT][ MSE = 0.2557 | LCC = 0.8599 | SRCC = 0.8582 ]
65
+ INFO:root:Average loss=0.16473962548069465
66
+ INFO:root:
67
+ [Valid][20][UTT][ MSE = 0.2713 | LCC = 0.8644 | SRCC = 0.8624 ]
68
+ INFO:root:Average loss=0.1695888197192779
69
+ INFO:root:
70
+ [Valid][21][UTT][ MSE = 0.2281 | LCC = 0.8591 | SRCC = 0.8583 ]
71
+ INFO:root:Average loss=0.15933551925879258
72
+ INFO:root:
73
+ [Valid][22][UTT][ MSE = 0.2526 | LCC = 0.8579 | SRCC = 0.8556 ]
74
+ INFO:root:Average loss=0.14816596855719885
75
+ INFO:root:
76
+ [Valid][23][UTT][ MSE = 0.2568 | LCC = 0.8595 | SRCC = 0.8589 ]
77
+ INFO:root:Average loss=0.15036036523106772
78
+ INFO:root:
79
+ [Valid][24][UTT][ MSE = 0.2616 | LCC = 0.8517 | SRCC = 0.8531 ]
80
+ INFO:root:Average loss=0.1405003054592854
81
+ INFO:root:
82
+ [Valid][25][UTT][ MSE = 0.2301 | LCC = 0.8589 | SRCC = 0.8591 ]
83
+ INFO:root:Average loss=0.140032014499108
84
+ INFO:root:
85
+ [Valid][26][UTT][ MSE = 0.2576 | LCC = 0.8561 | SRCC = 0.8565 ]
86
+ INFO:root:Average loss=0.1408207094631134
87
+ INFO:root:
88
+ [Valid][27][UTT][ MSE = 0.2308 | LCC = 0.8619 | SRCC = 0.8610 ]
89
+ INFO:root:Average loss=0.1366943697898816
90
+ INFO:root:
91
+ [Valid][28][UTT][ MSE = 0.2400 | LCC = 0.8593 | SRCC = 0.8582 ]
92
+ INFO:root:Average loss=0.13014958646052924
93
+ INFO:root:
94
+ [Valid][29][UTT][ MSE = 0.2415 | LCC = 0.8564 | SRCC = 0.8551 ]
95
+ INFO:root:Average loss=0.12681995341793084
96
+ INFO:root:
97
+ [Valid][30][UTT][ MSE = 0.2585 | LCC = 0.8520 | SRCC = 0.8501 ]
98
+ INFO:root:
99
+ [Test][31][UTT][ MSE = 0.2955 | LCC = 0.8546 | SRCC = 0.8541 ]
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/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_2epoch/layer9/run4/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc97282c85157499e35d31ae8323f0ebf4c60e9b69d16a3f7b9cf28675c87eeb
3
+ size 2747979
results_bvcc/w2v2_base_bvcc_finetuned_2epoch/layer9/run4/model_best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e48897cd2e4fd733ede66bbcaa4e114dbc7093a2593f07663faeb8f0f94226ed
3
+ size 2749702