Saving weights and logs of epoch 6

config.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1fa5adbbe6a5cbeedcab79760dd7611a074c1905f4f79d8d386f835eee730a9
-size 701
+oid sha256:ba2f2141e020e11d5a8378398ab356436905f898315d8ffe70b26dc91addc441
+size 700

events.out.tfevents.1630919923.t1v-n-1a0a7c50-w-0.3421924.3.v2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78b1af55372d9d5e839b4800985716d13bd89ce660e256adc83a1d5d0d898750
+size 40

events.out.tfevents.1630921213.t1v-n-1a0a7c50-w-0.3431412.3.v2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:442420b6e0a7477c4764844291689e084536f4a835cb79e27fea6333ee3931cb
+size 2865513

flax_model.msgpack

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5fe3762aac25de79ac89d75581bfcbc3bfd701c3f616afa4b7341e97c6414497
-size 1100762015
+oid sha256:5167e3435eacd785c79013187f090ebc1538ea5a04d3f649692cd929c470268c
+size 2329617315

run_summarization_t5.sh

@@ -1,18 +1,20 @@
 
 python run_summarization_t5_flax.py \
 	--output_dir "." \
-       	--model_name_or_path pere/norwegian-t5-base-NCC-fast \
-       	--tokenizer_name pere/norwegian-t5-base-NCC-fast \
+       	--model_name_or_path google/mt5-base \
+       	--tokenizer_name google/mt5-base \
 	--dataset_name  xsum \
 	--do_train --do_eval --do_predict \
 	--predict_with_generate \
 	--num_train_epochs 6 \
 	--learning_rate 5e-5 \
 	--warmup_steps 0 \
-	--per_device_train_batch_size 32 \
-	--per_device_eval_batch_size 32 \
+	--per_device_train_batch_size 8 \
+	--per_device_eval_batch_size 8 \
 	--overwrite_output_dir \
 	--max_source_length 512 \
 	--max_target_length 64 \
-	--revision $1 \
 	--push_to_hub
+
+
+#--revision None \

run_summarization_t5_flax.py

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4e2782269c57143454f219ab7563eaed06ab04782ddb7c7d50906b1f1302b4b3
-size 34784
+oid sha256:f7bb91fa2da0e303cda754fe06548c3710f06dcddf575795da18eca4e67976d5
+size 35053