Add results on evaluation data

README.md

@@ -8,19 +8,42 @@ tags:
 - generated_from_trainer
 - robust-speech-event
 model-index:
-- name: ''
-  results: []
+- name: XLS-R-1B - French
+  results:
+  - task: 
+      name: Automatic Speech Recognition 
+      type: automatic-speech-recognition
+    dataset:
+      name: Common Voice 8
+      type: mozilla-foundation/common_voice_8_0
+      args: fr
+    metrics:
+       - name: Test WER
+         type: wer
+         value: 18.33
+       - name: Test CER
+         type: cer
+         value: 5.60
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: Robust Speech Event - Dev Data
+      type: speech-recognition-community-v2/dev_data
+      args: fr
+    metrics:
+       - name: Test WER
+         type: wer
+         value: 60.25
+       - name: Test CER
+         type: cer
+         value: 15.68
 ---
 
 ## Model description
 
 This model is a fine-tuned version of [facebook/wav2vec2-xls-r-1b](https://huggingface.co/facebook/wav2vec2-xls-r-1b) on the MOZILLA-FOUNDATION/COMMON_VOICE_8_0 - FR dataset.
 
-## Training and evaluation data
-
-It achieves the following results on the evaluation set:
-- Wer: 0.1834 (STEP 13000)
-
 ## Training procedure
 
 ### Training hyperparameters
@@ -56,6 +79,9 @@ The following hyperparameters were used during training:
 | 0.8015        | 3.44  | 12000 | inf             | 0.1880 |
 | 0.774         | 3.73  | 13000 | inf             | 0.1834 |
 
+It achieves the best result for STEP 13000 on the validation set:
+- Wer: 0.1834
+
 Some problems occur when calculating the validation loss.
 
 ### Framework versions
@@ -64,3 +90,15 @@ Some problems occur when calculating the validation loss.
 - Pytorch 1.10.2+cu102
 - Datasets 1.18.3.dev0
 - Tokenizers 0.11.0
+
+### Evaluation Commands
+1. To evaluate on `mozilla-foundation/common_voice_8` with split `test`
+
+```bash
+python eval.py --model_id Plim/xls-r-1b-cv_8-fr --dataset mozilla-foundation/common_voice_8_0 --config fr --split test
+```
+
+2. To evaluate on `speech-recognition-community-v2/dev_data`
+
+```bash
+python eval.py --model_id Plim/xls-r-1b-cv_8-fr --dataset speech-recognition-community-v2/dev_data --config fr --split validation --chunk_length_s 5.0 --stride_length_s 1.0

mozilla-foundation_common_voice_8_0_fr_test_eval_results.txt

@@ -0,0 +1,2 @@
+WER: 0.18333515105245937
+CER: 0.05606368028384753

speech-recognition-community-v2_dev_data_fr_validation_eval_results.txt

@@ -0,0 +1,2 @@
+WER: 0.6024801374275285
+CER: 0.15675326436359865

wandb/run-20220203_170643-2fkfdtzb/files/config.yaml

@@ -9437,7 +9437,13 @@ _wandb:
       - 1
       - 5
       - 11
+      2:
+      - 1
+      - 5
+      - 11
       3:
+      - 1
+      - 7
       - 13
       4: 3.8.8
       5: 0.12.10

wandb/run-20220203_170643-2fkfdtzb/files/output.log

@@ -20888,3 +20888,12 @@ Upload file wandb/run-20220203_170643-2fkfdtzb/run-2fkfdtzb.wandb: 100%|██
 02/05/2022 20:05:15 - WARNING - huggingface_hub.repository - To https://huggingface.co/Plim/xls-r-1b-cv_8-fr
    33879a4..738d48b  main -> main
 Dropping the following result as it does not have all the necessary fields:
+{}
+
+
+
+Upload file wandb/run-20220203_170643-2fkfdtzb/run-2fkfdtzb.wandb:  97%|██████████████████████████████████████████████████████████████████▏ | 189M/194M [00:08<00:00, 28.8MB/s]To https://huggingface.co/Plim/xls-r-1b-cv_8-fr
+   738d48b..f6ce89f  main -> main
+Upload file wandb/run-20220203_170643-2fkfdtzb/run-2fkfdtzb.wandb: 100%|████████████████████████████████████████████████████████████████████| 194M/194M [00:09<00:00, 22.0MB/s]
+02/05/2022 20:05:40 - WARNING - huggingface_hub.repository - To https://huggingface.co/Plim/xls-r-1b-cv_8-fr
+   738d48b..f6ce89f  main -> main

wandb/run-20220203_170643-2fkfdtzb/logs/debug-internal.log

@@ -53927,3 +53927,128 @@
 2022-02-05 20:05:19,293 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: stop_status
 2022-02-05 20:05:19,294 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: stop_status
 2022-02-05 20:05:20,358 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:26,198 DEBUG   SenderThread:5297 [sender.py:send():235] send: stats
+2022-02-05 20:05:34,366 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:34,473 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: stop_status
+2022-02-05 20:05:34,474 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: stop_status
+2022-02-05 20:05:36,368 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:38,369 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:40,371 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:42,372 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:43,373 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:48,070 DEBUG   SenderThread:5297 [sender.py:send():235] send: telemetry
+2022-02-05 20:05:48,071 DEBUG   SenderThread:5297 [sender.py:send():235] send: exit
+2022-02-05 20:05:48,071 INFO    SenderThread:5297 [sender.py:send_exit():371] handling exit code: 0
+2022-02-05 20:05:48,072 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:48,072 INFO    SenderThread:5297 [sender.py:send_exit():373] handling runtime: 183542
+2022-02-05 20:05:48,128 INFO    SenderThread:5297 [sender.py:_save_file():944] saving file wandb-summary.json with policy end
+2022-02-05 20:05:48,129 INFO    SenderThread:5297 [sender.py:send_exit():379] send defer
+2022-02-05 20:05:48,129 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:48,130 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:48,131 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 0
+2022-02-05 20:05:48,131 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:48,131 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 0
+2022-02-05 20:05:48,131 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 1
+2022-02-05 20:05:48,132 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:48,132 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 1
+2022-02-05 20:05:48,160 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:48,160 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 1
+2022-02-05 20:05:48,160 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 2
+2022-02-05 20:05:48,160 DEBUG   SenderThread:5297 [sender.py:send():235] send: stats
+2022-02-05 20:05:48,161 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:48,161 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 2
+2022-02-05 20:05:48,162 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:48,162 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 2
+2022-02-05 20:05:48,162 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 3
+2022-02-05 20:05:48,162 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:48,163 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 3
+2022-02-05 20:05:48,228 DEBUG   SenderThread:5297 [sender.py:send():235] send: summary
+2022-02-05 20:05:48,238 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:48,319 INFO    SenderThread:5297 [sender.py:_save_file():944] saving file wandb-summary.json with policy end
+2022-02-05 20:05:48,319 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:48,319 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 3
+2022-02-05 20:05:48,319 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 4
+2022-02-05 20:05:48,319 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:48,321 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:48,321 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 4
+2022-02-05 20:05:48,321 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:48,321 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 4
+2022-02-05 20:05:48,377 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/wandb-summary.json
+2022-02-05 20:05:48,378 INFO    Thread-8  :5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:48,422 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,347 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 5
+2022-02-05 20:05:49,347 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:49,349 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:49,349 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 5
+2022-02-05 20:05:49,349 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:49,349 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 5
+2022-02-05 20:05:49,349 INFO    SenderThread:5297 [dir_watcher.py:finish():283] shutting down directory watcher
+2022-02-05 20:05:49,379 INFO    SenderThread:5297 [dir_watcher.py:_on_file_modified():230] file/dir modified: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/config.yaml
+2022-02-05 20:05:49,380 INFO    SenderThread:5297 [dir_watcher.py:finish():313] scan: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files
+2022-02-05 20:05:49,380 INFO    SenderThread:5297 [dir_watcher.py:finish():327] scan save: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/requirements.txt requirements.txt
+2022-02-05 20:05:49,380 INFO    SenderThread:5297 [dir_watcher.py:finish():327] scan save: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/config.yaml config.yaml
+2022-02-05 20:05:49,381 INFO    SenderThread:5297 [dir_watcher.py:finish():327] scan save: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/wandb-metadata.json wandb-metadata.json
+2022-02-05 20:05:49,381 INFO    SenderThread:5297 [dir_watcher.py:finish():327] scan save: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log output.log
+2022-02-05 20:05:49,387 INFO    SenderThread:5297 [dir_watcher.py:finish():327] scan save: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/conda-environment.yaml conda-environment.yaml
+2022-02-05 20:05:49,393 INFO    SenderThread:5297 [dir_watcher.py:finish():327] scan save: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/wandb-summary.json wandb-summary.json
+2022-02-05 20:05:49,394 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 6
+2022-02-05 20:05:49,402 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:49,402 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 6
+2022-02-05 20:05:49,403 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:49,403 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 6
+2022-02-05 20:05:49,403 INFO    SenderThread:5297 [file_pusher.py:finish():177] shutting down file pusher
+2022-02-05 20:05:49,450 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,451 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:49,553 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,554 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:49,657 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,657 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:49,760 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,760 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:49,863 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,863 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:49,965 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:49,966 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:50,035 INFO    Thread-12 :5297 [upload_job.py:push():137] Uploaded file /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/requirements.txt
+2022-02-05 20:05:50,069 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:50,069 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:50,172 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:50,173 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:50,197 INFO    Thread-14 :5297 [upload_job.py:push():137] Uploaded file /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/output.log
+2022-02-05 20:05:50,219 INFO    Thread-13 :5297 [upload_job.py:push():137] Uploaded file /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/config.yaml
+2022-02-05 20:05:50,225 INFO    Thread-15 :5297 [upload_job.py:push():137] Uploaded file /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/files/wandb-summary.json
+2022-02-05 20:05:50,276 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:50,277 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:50,380 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:50,380 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:50,426 INFO    Thread-7  :5297 [sender.py:transition_state():392] send defer: 7
+2022-02-05 20:05:50,427 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:50,427 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 7
+2022-02-05 20:05:50,428 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:50,428 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 7
+2022-02-05 20:05:50,482 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:51,532 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 8
+2022-02-05 20:05:51,533 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:51,534 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:51,534 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 8
+2022-02-05 20:05:51,535 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:51,535 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 8
+2022-02-05 20:05:51,535 INFO    SenderThread:5297 [sender.py:transition_state():392] send defer: 9
+2022-02-05 20:05:51,538 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: defer
+2022-02-05 20:05:51,540 INFO    HandlerThread:5297 [handler.py:handle_request_defer():154] handle defer: 9
+2022-02-05 20:05:51,540 DEBUG   SenderThread:5297 [sender.py:send():235] send: final
+2022-02-05 20:05:51,540 DEBUG   SenderThread:5297 [sender.py:send():235] send: footer
+2022-02-05 20:05:51,540 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: defer
+2022-02-05 20:05:51,540 INFO    SenderThread:5297 [sender.py:send_request_defer():388] handle sender defer: 9
+2022-02-05 20:05:51,636 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: poll_exit
+2022-02-05 20:05:51,637 DEBUG   SenderThread:5297 [sender.py:send_request():249] send_request: poll_exit
+2022-02-05 20:05:51,637 INFO    SenderThread:5297 [file_pusher.py:join():182] waiting for file pusher
+2022-02-05 20:05:51,944 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: get_summary
+2022-02-05 20:05:52,067 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: sampled_history
+2022-02-05 20:05:52,073 DEBUG   HandlerThread:5297 [handler.py:handle_request():131] handle_request: shutdown
+2022-02-05 20:05:52,073 INFO    HandlerThread:5297 [handler.py:finish():739] shutting down handler
+2022-02-05 20:05:52,538 INFO    WriterThread:5297 [datastore.py:close():281] close: /workspace/xls-r-1b-cv_8-fr/wandb/run-20220203_170643-2fkfdtzb/run-2fkfdtzb.wandb
+2022-02-05 20:05:52,942 INFO    SenderThread:5297 [sender.py:finish():1075] shutting down sender
+2022-02-05 20:05:52,942 INFO    SenderThread:5297 [file_pusher.py:finish():177] shutting down file pusher
+2022-02-05 20:05:52,942 INFO    SenderThread:5297 [file_pusher.py:join():182] waiting for file pusher
+2022-02-05 20:05:52,954 INFO    MainThread:5297 [internal.py:handle_exit():79] Internal process exited

wandb/run-20220203_170643-2fkfdtzb/logs/debug.log

@@ -24,3 +24,133 @@ config: {}
 2022-02-03 17:06:46,149 INFO    MainThread:43340 [wandb_init.py:init():651] run started, returning control to user process
 2022-02-03 17:06:46,152 INFO    MainThread:43340 [wandb_run.py:_config_callback():966] config_cb None None {'return_dict': True, 'output_hidden_states': False, 'output_attentions': False, 'torchscript': False, 'torch_dtype': 'float32', 'use_bfloat16': False, 'pruned_heads': {}, 'tie_word_embeddings': True, 'is_encoder_decoder': False, 'is_decoder': False, 'cross_attention_hidden_size': None, 'add_cross_attention': False, 'tie_encoder_decoder': False, 'max_length': 20, 'min_length': 0, 'do_sample': False, 'early_stopping': False, 'num_beams': 1, 'num_beam_groups': 1, 'diversity_penalty': 0.0, 'temperature': 1.0, 'top_k': 50, 'top_p': 1.0, 'repetition_penalty': 1.0, 'length_penalty': 1.0, 'no_repeat_ngram_size': 0, 'encoder_no_repeat_ngram_size': 0, 'bad_words_ids': None, 'num_return_sequences': 1, 'chunk_size_feed_forward': 0, 'output_scores': False, 'return_dict_in_generate': False, 'forced_bos_token_id': None, 'forced_eos_token_id': None, 'remove_invalid_values': False, 'architectures': ['Wav2Vec2ForPreTraining'], 'finetuning_task': None, 'id2label': {0: 'LABEL_0', 1: 'LABEL_1'}, 'label2id': {'LABEL_0': 0, 'LABEL_1': 1}, 'tokenizer_class': None, 'prefix': None, 'bos_token_id': 1, 'pad_token_id': 45, 'eos_token_id': 2, 'sep_token_id': None, 'decoder_start_token_id': None, 'task_specific_params': None, 'problem_type': None, '_name_or_path': 'facebook/wav2vec2-xls-r-1b', 'transformers_version': '4.17.0.dev0', 'feat_extract_dropout': 0.0, 'model_type': 'wav2vec2', 'num_feat_extract_layers': 7, 'hidden_size': 1280, 'feat_extract_norm': 'layer', 'feat_extract_activation': 'gelu', 'conv_dim': [512, 512, 512, 512, 512, 512, 512], 'conv_stride': [5, 2, 2, 2, 2, 2, 2], 'conv_kernel': [10, 3, 3, 3, 3, 2, 2], 'conv_bias': True, 'num_conv_pos_embeddings': 128, 'num_conv_pos_embedding_groups': 16, 'num_hidden_layers': 48, 'intermediate_size': 5120, 'hidden_act': 'gelu', 'num_attention_heads': 16, 'hidden_dropout': 0.0, 'attention_dropout': 0.0, 'activation_dropout': 0.1, 'feat_proj_dropout': 0.0, 'final_dropout': 0.0, 'layerdrop': 0.0, 'layer_norm_eps': 1e-05, 'initializer_range': 0.02, 'vocab_size': 46, 'do_stable_layer_norm': True, 'use_weighted_layer_sum': False, 'apply_spec_augment': True, 'mask_time_prob': 0.75, 'mask_time_length': 10, 'mask_time_min_masks': 2, 'mask_feature_prob': 0.25, 'mask_feature_length': 64, 'mask_feature_min_masks': 0, 'num_codevectors_per_group': 320, 'num_codevector_groups': 2, 'contrastive_logits_temperature': 0.1, 'feat_quantizer_dropout': 0.0, 'num_negatives': 100, 'codevector_dim': 1024, 'proj_codevector_dim': 1024, 'diversity_loss_weight': 0.1, 'ctc_loss_reduction': 'mean', 'ctc_zero_infinity': False, 'add_adapter': False, 'adapter_kernel_size': 3, 'adapter_stride': 2, 'num_adapter_layers': 3, 'output_hidden_size': 1280, 'classifier_proj_size': 256, 'tdnn_dim': [512, 512, 512, 512, 1500], 'tdnn_kernel': [5, 3, 3, 1, 1], 'tdnn_dilation': [1, 2, 3, 1, 1], 'xvector_output_dim': 512, 'output_dir': './', 'overwrite_output_dir': True, 'do_train': True, 'do_eval': True, 'do_predict': False, 'evaluation_strategy': 'steps', 'prediction_loss_only': False, 'per_device_train_batch_size': 16, 'per_device_eval_batch_size': 16, 'per_gpu_train_batch_size': 'None', 'per_gpu_eval_batch_size': 'None', 'gradient_accumulation_steps': 8, 'eval_accumulation_steps': 'None', 'learning_rate': 7.5e-05, 'weight_decay': 0.0, 'adam_beta1': 0.9, 'adam_beta2': 0.999, 'adam_epsilon': 1e-08, 'max_grad_norm': 1.0, 'num_train_epochs': 4.0, 'max_steps': -1, 'lr_scheduler_type': 'linear', 'warmup_ratio': 0.0, 'warmup_steps': 2000, 'log_level': -1, 'log_level_replica': -1, 'log_on_each_node': True, 'logging_dir': './runs/Feb03_13-28-41_job-597becdf-05fc-498e-bdc5-d363b0af8ddd', 'logging_strategy': 'steps', 'logging_first_step': False, 'logging_steps': 100, 'logging_nan_inf_filter': True, 'save_strategy': 'steps', 'save_steps': 1000, 'save_total_limit': 3, 'save_on_each_node': False, 'no_cuda': False, 'seed': 42, 'bf16': False, 'fp16': True, 'fp16_opt_level': 'O1', 'half_precision_backend': 'amp', 'bf16_full_eval': False, 'fp16_full_eval': False, 'tf32': 'None', 'local_rank': -1, 'xpu_backend': 'None', 'tpu_num_cores': 'None', 'tpu_metrics_debug': False, 'debug': '[]', 'dataloader_drop_last': False, 'eval_steps': 1000, 'dataloader_num_workers': 0, 'past_index': -1, 'run_name': './', 'disable_tqdm': False, 'remove_unused_columns': True, 'label_names': 'None', 'load_best_model_at_end': True, 'metric_for_best_model': 'loss', 'greater_is_better': False, 'ignore_data_skip': False, 'sharded_ddp': '[]', 'deepspeed': 'None', 'label_smoothing_factor': 0.0, 'optim': 'adamw_hf', 'adafactor': False, 'group_by_length': True, 'length_column_name': 'input_length', 'report_to': "['wandb']", 'ddp_find_unused_parameters': 'None', 'ddp_bucket_cap_mb': 'None', 'dataloader_pin_memory': True, 'skip_memory_metrics': True, 'use_legacy_prediction_loop': False, 'push_to_hub': True, 'resume_from_checkpoint': 'None', 'hub_model_id': 'None', 'hub_strategy': 'every_save', 'hub_token': '<HUB_TOKEN>', 'gradient_checkpointing': True, 'fp16_backend': 'auto', 'push_to_hub_model_id': 'None', 'push_to_hub_organization': 'None', 'push_to_hub_token': '<PUSH_TO_HUB_TOKEN>', '_n_gpu': 1, 'mp_parameters': '', 'train_batch_size': 16, 'eval_batch_size': 16}
 2022-02-03 17:06:46,160 INFO    MainThread:43340 [wandb_watch.py:watch():43] Watching
+2022-02-05 20:05:45,327 INFO    MainThread:43340 [wandb_run.py:_atexit_cleanup():1797] got exitcode: 0
+2022-02-05 20:05:45,332 INFO    MainThread:43340 [wandb_run.py:_restore():1769] restore
+2022-02-05 20:05:48,131 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 1
+}
+pusher_stats {
+  uploaded_bytes: 2037
+  total_bytes: 2037
+}
+
+2022-02-05 20:05:48,321 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 1
+}
+pusher_stats {
+  uploaded_bytes: 2037
+  total_bytes: 2037
+}
+
+2022-02-05 20:05:49,349 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 1
+}
+pusher_stats {
+  uploaded_bytes: 2037
+  total_bytes: 2037
+}
+
+2022-02-05 20:05:49,452 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 2037
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:49,555 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 2037
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:49,658 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 116725
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:49,762 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:49,864 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:49,967 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:50,071 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:50,175 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:50,278 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:50,381 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:51,534 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+
+2022-02-05 20:05:51,943 INFO    MainThread:43340 [wandb_run.py:_wait_for_finish():1929] got exit ret: done: true
+exit_result {
+}
+file_counts {
+  wandb_count: 5
+}
+pusher_stats {
+  uploaded_bytes: 1833587
+  total_bytes: 1833587
+}
+local_info {
+}
+
+2022-02-05 20:05:53,125 INFO    MainThread:43340 [wandb_run.py:_append_history():2144] rendering history
+2022-02-05 20:05:53,127 INFO    MainThread:43340 [wandb_run.py:_append_summary():2102] rendering summary
+2022-02-05 20:05:53,129 INFO    MainThread:43340 [wandb_run.py:_append_files():2194] logging synced files

wandb/run-20220203_170643-2fkfdtzb/run-2fkfdtzb.wandb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:26a4a45f21c2571d76ee29bd77a04d51719a8ddc69f771470e04b1d33129aab1
-size 203570905
+oid sha256:aa27b93a052e1dc7abbf9471758aac62025a541fccb4f39477aa762b4b927865
+size 205039935