M9 models

.gitattributes

@@ -30,3 +30,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,23 @@
+# mBART based translation model
+model_path: /data/scratch/haukur/document_translation/checkpoints/v2.all.wordnoise0.06.fragmentnoise0.06.lr8e-06.dropout0.1.spmalpha0.7.seed228/checkpoint_best.pt
+num_updates: 14500
+Source language: en_XX
+Target language: is_IS
+
+---
+tags:
+- translation
+inference:
+  parameters:
+    src_lang: "is_IS"
+    tgt_lang: "en_XX"
+    decoder_start_token_id: 250004
+    max_length: 512
+widget:
+- text: "Ég ætti að kaupa bát."
+language:
+- en
+- is
+datasets:
+- train.fornsogur,train.greynir_articles,train.greynir_articles_2021,train.hirslan,train.ic3_filtered,train.rafbokavefurinn,train.rmh_filtered,train.wikipedia,train.abstracts,train.studentabladid,train.jw300,train.rannsoknarskyrsla_althingis,train.eea,train.fornsogur,train.greynir_articles,train.greynir_articles_2021,train.hirslan,train.ic3_filtered,train.rafbokavefurinn,train.rmh_filtered,train.wikipedia,train.bible
+---

config.json

@@ -0,0 +1,34 @@
+{
+  "activation_dropout": 0.0,
+  "activation_function": "gelu",
+  "architectures": [
+    "MBartForConditionalGeneration"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 0,
+  "classifier_dropout": 0.0,
+  "d_model": 1024,
+  "decoder_attention_heads": 16,
+  "decoder_ffn_dim": 4096,
+  "decoder_layerdrop": 0.0,
+  "decoder_layers": 12,
+  "dropout": 0.1,
+  "encoder_attention_heads": 16,
+  "encoder_ffn_dim": 4096,
+  "encoder_layerdrop": 0.0,
+  "encoder_layers": 12,
+  "eos_token_id": 2,
+  "forced_eos_token_id": 2,
+  "init_std": 0.02,
+  "is_encoder_decoder": true,
+  "max_length": 1024,
+  "max_position_embeddings": 1024,
+  "model_type": "mbart",
+  "num_hidden_layers": 12,
+  "pad_token_id": 1,
+  "scale_embedding": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.22.2",
+  "use_cache": true,
+  "vocab_size": 250028
+}

pytorch_model.bin

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

special_tokens_map.json

@@ -0,0 +1,42 @@
+{
+  "additional_special_tokens": [
+    "ar_AR",
+    "cs_CZ",
+    "de_DE",
+    "en_XX",
+    "es_XX",
+    "et_EE",
+    "fi_FI",
+    "fr_XX",
+    "gu_IN",
+    "hi_IN",
+    "it_IT",
+    "is_IS",
+    "kk_KZ",
+    "ko_KR",
+    "lt_LT",
+    "lv_LV",
+    "my_MM",
+    "ne_NP",
+    "nl_XX",
+    "ro_RO",
+    "ru_RU",
+    "si_LK",
+    "tr_TR",
+    "vi_VN",
+    "zh_CN"
+  ],
+  "bos_token": "<s>",
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "mask_token": {
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<pad>",
+  "sep_token": "</s>",
+  "unk_token": "<unk>"
+}

tokenizer.json

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

tokenizer_config.json

@@ -0,0 +1,23 @@
+{
+  "additional_special_tokens": null,
+  "bos_token": "<s>",
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "mask_token": {
+    "__type": "AddedToken",
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "model_max_length": 1024,
+  "name_or_path": "mideind/tokenizer-mbart-25-enis",
+  "pad_token": "<pad>",
+  "sep_token": "</s>",
+  "special_tokens_map_file": null,
+  "src_lang": "en_XX",
+  "tgt_lang": "is_IS",
+  "tokenizer_class": "MBartTokenizer",
+  "unk_token": "<unk>"
+}

train_args.json

@@ -0,0 +1,198 @@
+{
+    "no_progress_bar": true,
+    "log_interval": 5,
+    "log_format": "simple",
+    "tensorboard_logdir": "/data/scratch/haukur/document_translation/tensorboard.logs/v2.all.wordnoise0.06.fragmentnoise0.06.lr8e-06.dropout0.1.spmalpha0.7.seed228",
+    "seed": 228,
+    "cpu": false,
+    "tpu": false,
+    "bf16": false,
+    "memory_efficient_bf16": false,
+    "fp16": true,
+    "memory_efficient_fp16": false,
+    "fp16_no_flatten_grads": false,
+    "fp16_init_scale": 128,
+    "fp16_scale_window": null,
+    "fp16_scale_tolerance": 0.0,
+    "min_loss_scale": 0.0001,
+    "threshold_loss_scale": null,
+    "user_dir": "/data/scratch/haukur/document_translation/fairseq_user_dir",
+    "empty_cache_freq": 0,
+    "all_gather_list_size": 16384,
+    "model_parallel_size": 1,
+    "checkpoint_suffix": "",
+    "checkpoint_shard_count": 1,
+    "quantization_config_path": null,
+    "profile": false,
+    "criterion": "cross_entropy",
+    "optimizer": "adam",
+    "lr_scheduler": "polynomial_decay",
+    "tokenizer": null,
+    "bpe": "sentencepiece",
+    "scoring": "bleu",
+    "task": "document_translation_from_pretrained_bart",
+    "num_workers": 8,
+    "skip_invalid_size_inputs_valid_test": true,
+    "max_tokens": 2000,
+    "batch_size": null,
+    "required_batch_size_multiple": 8,
+    "required_seq_len_multiple": 1,
+    "dataset_impl": null,
+    "data_buffer_size": 10,
+    "train_subset": "train.fornsogur,train.greynir_articles,train.greynir_articles_2021,train.hirslan,train.ic3_filtered,train.rafbokavefurinn,train.rmh_filtered,train.wikipedia,train.abstracts,train.studentabladid,train.jw300,train.rannsoknarskyrsla_althingis,train.eea,train.fornsogur,train.greynir_articles,train.greynir_articles_2021,train.hirslan,train.ic3_filtered,train.rafbokavefurinn,train.rmh_filtered,train.wikipedia,train.bible",
+    "valid_subset": "valid.newsdev2021,valid.flores_dev",
+    "validate_interval": 1,
+    "validate_interval_updates": 500,
+    "validate_after_updates": 0,
+    "fixed_validation_seed": null,
+    "disable_validation": false,
+    "max_tokens_valid": 2000,
+    "batch_size_valid": null,
+    "curriculum": 0,
+    "gen_subset": "test",
+    "num_shards": 1,
+    "shard_id": 0,
+    "distributed_world_size": 2,
+    "distributed_rank": 0,
+    "distributed_backend": "nccl",
+    "distributed_init_method": "tcp://localhost:18931",
+    "distributed_port": -1,
+    "device_id": 0,
+    "distributed_no_spawn": false,
+    "ddp_backend": "no_c10d",
+    "bucket_cap_mb": 25,
+    "fix_batches_to_gpus": false,
+    "find_unused_parameters": false,
+    "fast_stat_sync": false,
+    "broadcast_buffers": false,
+    "distributed_wrapper": "DDP",
+    "slowmo_momentum": null,
+    "slowmo_algorithm": "LocalSGD",
+    "localsgd_frequency": 3,
+    "nprocs_per_node": 2,
+    "pipeline_model_parallel": false,
+    "pipeline_balance": null,
+    "pipeline_devices": null,
+    "pipeline_chunks": 0,
+    "pipeline_encoder_balance": null,
+    "pipeline_encoder_devices": null,
+    "pipeline_decoder_balance": null,
+    "pipeline_decoder_devices": null,
+    "pipeline_checkpoint": "never",
+    "zero_sharding": "none",
+    "arch": "mbart_large",
+    "max_epoch": 0,
+    "max_update": 15000,
+    "stop_time_hours": 0,
+    "clip_norm": 3.0,
+    "sentence_avg": false,
+    "update_freq": [
+        10
+    ],
+    "lr": [
+        8e-06
+    ],
+    "min_lr": -1.0,
+    "use_bmuf": false,
+    "save_dir": "/data/scratch/haukur/document_translation/checkpoints/v2.all.wordnoise0.06.fragmentnoise0.06.lr8e-06.dropout0.1.spmalpha0.7.seed228",
+    "restore_file": "/data/scratch/haukur/document_translation/checkpoints/all.wordnoise0.06.fragmentnoise0.06.lr1e-05.dropout0.1.spmalpha0.7.seed226/checkpoint_best.pt",
+    "finetune_from_model": null,
+    "reset_dataloader": true,
+    "reset_lr_scheduler": true,
+    "reset_meters": true,
+    "reset_optimizer": true,
+    "optimizer_overrides": "{}",
+    "save_interval": 1,
+    "save_interval_updates": 500,
+    "keep_interval_updates": 2,
+    "keep_last_epochs": -1,
+    "keep_best_checkpoints": -1,
+    "no_save": false,
+    "no_epoch_checkpoints": false,
+    "no_last_checkpoints": false,
+    "no_save_optimizer_state": false,
+    "best_checkpoint_metric": "ppl",
+    "maximize_best_checkpoint_metric": false,
+    "patience": 10,
+    "no_token_positional_embeddings": false,
+    "no_cross_attention": false,
+    "cross_self_attention": false,
+    "encoder_layerdrop": 0,
+    "decoder_layerdrop": 0,
+    "encoder_layers_to_keep": null,
+    "decoder_layers_to_keep": null,
+    "quant_noise_pq": 0,
+    "quant_noise_pq_block_size": 8,
+    "quant_noise_scalar": 0,
+    "adam_betas": "(0.9, 0.98)",
+    "adam_eps": 1e-06,
+    "weight_decay": 0.01,
+    "use_old_adam": false,
+    "force_anneal": null,
+    "warmup_updates": 300,
+    "end_learning_rate": 0.0,
+    "power": 1.0,
+    "total_num_update": 1000000,
+    "sentencepiece_model": "/data/models/mbart25-cont-enis/sentence.bpe.model",
+    "data": "/data/scratch/haukur/document_translation/data",
+    "source_lang": "en_XX",
+    "target_lang": "is_IS",
+    "load_alignments": false,
+    "left_pad_source": true,
+    "left_pad_target": false,
+    "max_source_positions": 1024,
+    "max_target_positions": 1024,
+    "upsample_primary": 1,
+    "truncate_source": false,
+    "num_batch_buckets": 0,
+    "eval_bleu": false,
+    "eval_bleu_detok": "space",
+    "eval_bleu_detok_args": null,
+    "eval_tokenized_bleu": false,
+    "eval_bleu_remove_bpe": null,
+    "eval_bleu_args": null,
+    "eval_bleu_print_samples": false,
+    "langs": "ar_AR,cs_CZ,de_DE,en_XX,es_XX,et_EE,fi_FI,fr_XX,gu_IN,hi_IN,it_IT,is_IS,ja_XX,kk_KZ,ko_KR,lt_LT,lv_LV,my_MM,ne_NP,nl_XX,ro_RO,ru_RU,si_LK,tr_TR,vi_VN,zh_CN",
+    "prepend_bos": false,
+    "max_sentences": 300,
+    "bt_subset": "train.fornsogur,train.greynir_articles,train.greynir_articles_2021,train.hirslan,train.ic3_filtered,train.rafbokavefurinn,train.rmh_filtered,train.wikipedia",
+    "align_subset": "train.abstracts,train.studentabladid,train.jw300,train.rannsoknarskyrsla_althingis,train.eea,train.fornsogur,train.greynir_articles,train.greynir_articles_2021,train.hirslan,train.ic3_filtered,train.rafbokavefurinn,train.rmh_filtered,train.wikipedia,train.bible",
+    "sentencepiece_alpha": 1.0,
+    "parallel_prob": 0.33,
+    "word_noise_prob": 0.06,
+    "fragment_noise_prob": 0.06,
+    "max_shuffle_dist": 3,
+    "encoder_normalize_before": true,
+    "decoder_normalize_before": true,
+    "layernorm_embedding": true,
+    "dropout": 0.1,
+    "attention_dropout": 0.1,
+    "no_seed_provided": false,
+    "no_scale_embedding": false,
+    "encoder_embed_path": null,
+    "encoder_embed_dim": 1024,
+    "encoder_ffn_embed_dim": 4096,
+    "encoder_layers": 12,
+    "encoder_attention_heads": 16,
+    "encoder_learned_pos": true,
+    "decoder_embed_path": null,
+    "decoder_embed_dim": 1024,
+    "decoder_ffn_embed_dim": 4096,
+    "decoder_layers": 12,
+    "decoder_attention_heads": 16,
+    "decoder_learned_pos": true,
+    "relu_dropout": 0.0,
+    "adaptive_softmax_cutoff": null,
+    "adaptive_softmax_dropout": 0,
+    "share_decoder_input_output_embed": true,
+    "share_all_embeddings": true,
+    "decoder_output_dim": 1024,
+    "decoder_input_dim": 1024,
+    "activation_fn": "gelu",
+    "pooler_activation_fn": "tanh",
+    "pooler_dropout": 0.0,
+    "distributed_num_procs": 2,
+    "activation_dropout": 0.0,
+    "adaptive_input": false,
+    "tie_adaptive_weights": false
+}