Saving weights and logs of epoch 0

config.json

@@ -1,7 +1,9 @@
 {
+  "_name_or_path": "./",
   "architectures": [
     "SpeechEncoderDecoderModel"
   ],
+  "cache_dir": "/home/sanchit_huggingface_co/cache/huggingface/transformers",
   "decoder": {
     "_name_or_path": "facebook/bart-large-cnn",
     "_num_labels": 3,

create_model.py

@@ -4,7 +4,7 @@ from transformers import AutoFeatureExtractor, AutoTokenizer, FlaxSpeechEncoderD
 encoder_id = "facebook/wav2vec2-large-lv60"
 decoder_id = "facebook/bart-large-cnn"
 
-model = FlaxSpeechEncoderDecoderModel.from_encoder_decoder_pretrained(encoder_id, decoder_id, encoder_add_adapter=True)
+model = FlaxSpeechEncoderDecoderModel.from_encoder_decoder_pretrained(encoder_id, decoder_id, encoder_add_adapter=True, cache_dir="/home/sanchit_huggingface_co/cache/huggingface/transormers")
 
 model.config.encoder.feat_proj_dropout = 0.0
 model.config.encoder.final_dropout = 0.0

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:59f268b5fa61d5eb20929dce97dfcfbd6123017d61789aef700566461af24242
+oid sha256:932441fd4d1167a741bd188a1f0bc953a229e065d97f15027e824bb6dbe4ae7b
 size 2353635949

run_librispeech.sh

@@ -6,6 +6,8 @@ python run_flax_speech_recognition_seq2seq.py \
         --train_split_name="train.100" \
         --eval_split_name="validation" \
         --output_dir="./" \
+        --dataset_cache_dir="/home/sanchit_huggingface_co/cache/huggingface/datasets" \
+        --cache_dir="/home/sanchit_huggingface_co/cache/huggingface/transformers" \
         --preprocessing_num_workers="16" \
         --length_column_name="input_length" \
         --overwrite_output_dir \

tokenizer_config.json

@@ -1 +1 @@
-{"errors": "replace", "bos_token": "<s>", "eos_token": "</s>", "sep_token": "</s>", "cls_token": "<s>", "unk_token": "<unk>", "pad_token": "<pad>", "mask_token": "<mask>", "add_prefix_space": false, "trim_offsets": true, "model_max_length": 1024, "special_tokens_map_file": null, "name_or_path": "facebook/bart-large-cnn", "tokenizer_class": "BartTokenizer"}
+{"errors": "replace", "bos_token": "<s>", "eos_token": "</s>", "sep_token": "</s>", "cls_token": "<s>", "unk_token": "<unk>", "pad_token": "<pad>", "mask_token": "<mask>", "add_prefix_space": false, "trim_offsets": true, "model_max_length": 1024, "special_tokens_map_file": null, "name_or_path": "./", "tokenizer_class": "BartTokenizer"}