checkpoints/script_1/0000040000/train_state_00007.json

{"step": 40000, "acc_step": 0, "data_loader_state": {"it_state": {"start_token": 6895, "it_state": {"it_state": {"root_dir": "/scratch/craffel/lingua/data/flexitok/", "sources": {"fw_edu": 0.4, "dan_Latn": 0.0216582869670702, "swe_Latn": 0.0216359765418466, "vie_Latn": 0.0197485510268674, "hun_Latn": 0.0247194573562308, "fas_Arab": 0.0205634624231076, "tur_Latn": 0.0235455794841729, "ces_Latn": 0.0248024455266208, "arb_Arab": 0.0234323706569333, "ell_Grek": 0.0233670886888026, "ind_Latn": 0.0269322054593488, "nld_Latn": 0.0277796326621489, "pol_Latn": 0.0294120104572311, "por_Latn": 0.0301413168306825, "ita_Latn": 0.0324056371021865, "jpn_Jpan": 0.03553104151369, "fra_Latn": 0.0381835560678536, "spa_Latn": 0.0387222793083669, "deu_Latn": 0.0419925340453022, "cmn_Hani": 0.0454067521384114, "rus_Cyrl": 0.0500198157431261}, "source_to_state": {"fw_edu": {"file_path": "/scratch/craffel/lingua/data/flexitok/fw_edu/fineweb_edu_100bt.chunk.07.jsonl", "position": 9511133535, "block_size": 1, "offset": 0, "current_iter": 0}, "dan_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/dan_Latn/fineweb_2_hq.dan_Latn.chunk.07.jsonl", "position": 442105065, "block_size": 1, "offset": 0, "current_iter": 0}, "swe_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/swe_Latn/fineweb_2_hq.swe_Latn.chunk.07.jsonl", "position": 411136309, "block_size": 1, "offset": 0, "current_iter": 0}, "vie_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/vie_Latn/fineweb_2_hq.vie_Latn.chunk.07.jsonl", "position": 438665701, "block_size": 1, "offset": 0, "current_iter": 0}, "hun_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/hun_Latn/fineweb_2_hq.hun_Latn.chunk.07.jsonl", "position": 580768793, "block_size": 1, "offset": 0, "current_iter": 0}, "fas_Arab": {"file_path": "/scratch/craffel/lingua/data/flexitok/fas_Arab/fineweb_2_hq.fas_Arab.chunk.07.jsonl", "position": 491380101, "block_size": 1, "offset": 0, "current_iter": 0}, "tur_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/tur_Latn/fineweb_2_hq.tur_Latn.chunk.07.jsonl", "position": 371424647, "block_size": 1, "offset": 0, "current_iter": 0}, "ces_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/ces_Latn/fineweb_2_hq.ces_Latn.chunk.07.jsonl", "position": 566029667, "block_size": 1, "offset": 0, "current_iter": 0}, "arb_Arab": {"file_path": "/scratch/craffel/lingua/data/flexitok/arb_Arab/fineweb_2_hq.arb_Arab.chunk.07.jsonl", "position": 643282760, "block_size": 1, "offset": 0, "current_iter": 0}, "ell_Grek": {"file_path": "/scratch/craffel/lingua/data/flexitok/ell_Grek/fineweb_2_hq.ell_Grek.chunk.07.jsonl", "position": 800256769, "block_size": 1, "offset": 0, "current_iter": 0}, "ind_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/ind_Latn/fineweb_2_hq.ind_Latn.chunk.07.jsonl", "position": 561773594, "block_size": 1, "offset": 0, "current_iter": 0}, "nld_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/nld_Latn/fineweb_2_hq.nld_Latn.chunk.07.jsonl", "position": 452523260, "block_size": 1, "offset": 0, "current_iter": 0}, "pol_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/pol_Latn/fineweb_2_hq.pol_Latn.chunk.07.jsonl", "position": 477379358, "block_size": 1, "offset": 0, "current_iter": 0}, "por_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/por_Latn/fineweb_2_hq.por_Latn.chunk.07.jsonl", "position": 489548509, "block_size": 1, "offset": 0, "current_iter": 0}, "ita_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/ita_Latn/fineweb_2_hq.ita_Latn.chunk.07.jsonl", "position": 546297574, "block_size": 1, "offset": 0, "current_iter": 0}, "jpn_Jpan": {"file_path": "/scratch/craffel/lingua/data/flexitok/jpn_Jpan/fineweb_2_hq.jpn_Jpan.chunk.07.jsonl", "position": 577089463, "block_size": 1, "offset": 0, "current_iter": 0}, "fra_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/fra_Latn/fineweb_2_hq.fra_Latn.chunk.07.jsonl", "position": 770830005, "block_size": 1, "offset": 0, "current_iter": 0}, "spa_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/spa_Latn/fineweb_2_hq.spa_Latn.chunk.07.jsonl", "position": 691529121, "block_size": 1, "offset": 0, "current_iter": 0}, "deu_Latn": {"file_path": "/scratch/craffel/lingua/data/flexitok/deu_Latn/fineweb_2_hq.deu_Latn.chunk.07.jsonl", "position": 828081558, "block_size": 1, "offset": 0, "current_iter": 0}, "cmn_Hani": {"file_path": "/scratch/craffel/lingua/data/flexitok/cmn_Hani/fineweb_2_hq.cmn_Hani.chunk.07.jsonl", "position": 880032722, "block_size": 1, "offset": 0, "current_iter": 0}, "rus_Cyrl": {"file_path": "/scratch/craffel/lingua/data/flexitok/rus_Cyrl/fineweb_2_hq.rus_Cyrl.chunk.07.jsonl", "position": 2058948331, "block_size": 1, "offset": 0, "current_iter": 0}}, "rng_state": {"bit_generator": "PCG64", "state": {"state": 135119677349206958523480063390548986231, "inc": 53245743019587277358203950863334653629}, "has_uint32": 0, "uinteger": 0}}, "add_bos": true, "add_eos": true, "name": "supertokenizer", "path": "meta-llama/Llama-3.2-1B", "tokenizers": [{"name": "huggingface", "path": "flexitok/bpe_script_Arab_16000", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_script_CmJp_16000", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_ltr_ell_Grek_8000_v2", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_ltr_fw_edu_32000_v2", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_ltr_hun_Latn_8000_v2", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_ltr_rus_Cyrl_16000_v2", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_ltr_tur_Latn_8000_v2", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_script_Germ_32000", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_script_Roma_32000", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_script_SEAS_16000", "load_supermapping": true}, {"name": "huggingface", "path": "flexitok/bpe_script_Slav_16000", "load_supermapping": true}], "dropout": 0.0, "rng_state": {"bit_generator": "PCG64", "state": {"state": 309660605569820076035861180006329659692, "inc": 332724090758049132448979897138935081983}, "has_uint32": 0, "uinteger": 4246645494}, "seed": 42, "superset_code_name": "script_1", "n_words": 165022, "routing": {"source_to_tokenizer": {"arb_Arab": "flexitok/bpe_script_Arab_16000", "fas_Arab": "flexitok/bpe_script_Arab_16000", "cmn_Hani": "flexitok/bpe_script_CmJp_16000", "jpn_Jpan": "flexitok/bpe_script_CmJp_16000", "ell_Grek": "flexitok/bpe_ltr_ell_Grek_8000_v2", "fw_edu": "flexitok/bpe_ltr_fw_edu_32000_v2", "hun_Latn": "flexitok/bpe_ltr_hun_Latn_8000_v2", "rus_Cyrl": "flexitok/bpe_ltr_rus_Cyrl_16000_v2", "tur_Latn": "flexitok/bpe_ltr_tur_Latn_8000_v2", "dan_Latn": "flexitok/bpe_script_Germ_32000", "deu_Latn": "flexitok/bpe_script_Germ_32000", "nld_Latn": "flexitok/bpe_script_Germ_32000", "swe_Latn": "flexitok/bpe_script_Germ_32000", "fra_Latn": "flexitok/bpe_script_Roma_32000", "ita_Latn": "flexitok/bpe_script_Roma_32000", "por_Latn": "flexitok/bpe_script_Roma_32000", "spa_Latn": "flexitok/bpe_script_Roma_32000", "ind_Latn": "flexitok/bpe_script_SEAS_16000", "vie_Latn": "flexitok/bpe_script_SEAS_16000", "ces_Latn": "flexitok/bpe_script_Slav_16000", "pol_Latn": "flexitok/bpe_script_Slav_16000"}, "suitable_tokenizer_probability": 0.9}}, "output_seq_len": 4096, "n_views": 2}, "seq_idx": 512, "rng_state": {"bit_generator": "PCG64", "state": {"state": 56948399199925506292066257204519406172, "inc": 19761753544780285878460645500694854795}, "has_uint32": 0, "uinteger": 1191791486}, "batch_size": 4, "prefetch_size": 1024}, "scheduler": {"base_lrs": [0.001], "last_epoch": 40000, "verbose": false, "_step_count": 40001, "_get_lr_called_within_step": false, "_last_lr": [0.0006726828545281266], "lr_lambdas": [{}]}}