Text Generation
Transformers
Safetensors
English
qwen3
code-agent
critic
process-reward-model
swe-bench
llama-factory
conversational
text-generation-inference
Instructions to use code-critic-model/Qwen3-4B-Critic-SFT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use code-critic-model/Qwen3-4B-Critic-SFT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="code-critic-model/Qwen3-4B-Critic-SFT") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("code-critic-model/Qwen3-4B-Critic-SFT") model = AutoModelForCausalLM.from_pretrained("code-critic-model/Qwen3-4B-Critic-SFT", device_map="auto") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=256) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use code-critic-model/Qwen3-4B-Critic-SFT with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "code-critic-model/Qwen3-4B-Critic-SFT" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "code-critic-model/Qwen3-4B-Critic-SFT", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/code-critic-model/Qwen3-4B-Critic-SFT
- SGLang
How to use code-critic-model/Qwen3-4B-Critic-SFT with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "code-critic-model/Qwen3-4B-Critic-SFT" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "code-critic-model/Qwen3-4B-Critic-SFT", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "code-critic-model/Qwen3-4B-Critic-SFT" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "code-critic-model/Qwen3-4B-Critic-SFT", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use code-critic-model/Qwen3-4B-Critic-SFT with Docker Model Runner:
docker model run hf.co/code-critic-model/Qwen3-4B-Critic-SFT
Download trainer_state.json from code-critic-model/Qwen3-4B-Critic-SFT: direct link, hf CLI and curl.
- Browser
- Download file 45.7 kB
-
https://huggingface.co/code-critic-model/Qwen3-4B-Critic-SFT/resolve/main/trainer_state.json
- Command line
-
hf download hf://code-critic-model/Qwen3-4B-Critic-SFT/trainer_state.json
-
curl -L -o trainer_state.json https://huggingface.co/code-critic-model/Qwen3-4B-Critic-SFT/resolve/main/trainer_state.json
45.7 kB
| { | |
| "best_global_step": null, | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "eval_steps": 500, | |
| "global_step": 2418, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.01240694789081886, | |
| "grad_norm": 24.380937576293945, | |
| "learning_rate": 1.8595041322314052e-07, | |
| "loss": 1.1434167861938476, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.02481389578163772, | |
| "grad_norm": 21.075366973876953, | |
| "learning_rate": 3.925619834710744e-07, | |
| "loss": 1.0735555648803712, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.03722084367245657, | |
| "grad_norm": 16.02638053894043, | |
| "learning_rate": 5.991735537190084e-07, | |
| "loss": 0.9607119560241699, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.04962779156327544, | |
| "grad_norm": 10.198869705200195, | |
| "learning_rate": 8.057851239669421e-07, | |
| "loss": 0.8850795745849609, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.062034739454094295, | |
| "grad_norm": 3.556870222091675, | |
| "learning_rate": 1.012396694214876e-06, | |
| "loss": 0.718876838684082, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.07444168734491315, | |
| "grad_norm": 4.056130886077881, | |
| "learning_rate": 1.21900826446281e-06, | |
| "loss": 0.6400353908538818, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.08684863523573201, | |
| "grad_norm": 3.1343917846679688, | |
| "learning_rate": 1.425619834710744e-06, | |
| "loss": 0.6471079826354981, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.09925558312655088, | |
| "grad_norm": 3.1576831340789795, | |
| "learning_rate": 1.632231404958678e-06, | |
| "loss": 0.6312197685241699, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.11166253101736973, | |
| "grad_norm": 3.7458317279815674, | |
| "learning_rate": 1.8388429752066117e-06, | |
| "loss": 0.6425768375396729, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.12406947890818859, | |
| "grad_norm": 3.5856683254241943, | |
| "learning_rate": 2.0454545454545457e-06, | |
| "loss": 0.5865562438964844, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.13647642679900746, | |
| "grad_norm": 3.186936616897583, | |
| "learning_rate": 2.2520661157024796e-06, | |
| "loss": 0.5968480110168457, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.1488833746898263, | |
| "grad_norm": 2.963216781616211, | |
| "learning_rate": 2.4586776859504134e-06, | |
| "loss": 0.6055496215820313, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.16129032258064516, | |
| "grad_norm": 3.0101115703582764, | |
| "learning_rate": 2.6652892561983473e-06, | |
| "loss": 0.5692682266235352, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.17369727047146402, | |
| "grad_norm": 2.968893527984619, | |
| "learning_rate": 2.871900826446281e-06, | |
| "loss": 0.5657201290130616, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.18610421836228289, | |
| "grad_norm": 3.103032112121582, | |
| "learning_rate": 3.078512396694215e-06, | |
| "loss": 0.5565766334533692, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.19851116625310175, | |
| "grad_norm": 3.6906442642211914, | |
| "learning_rate": 3.2851239669421493e-06, | |
| "loss": 0.5623440742492676, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.2109181141439206, | |
| "grad_norm": 3.6191394329071045, | |
| "learning_rate": 3.491735537190083e-06, | |
| "loss": 0.5390286445617676, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.22332506203473945, | |
| "grad_norm": 3.119192600250244, | |
| "learning_rate": 3.6983471074380165e-06, | |
| "loss": 0.504794979095459, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.23573200992555832, | |
| "grad_norm": 3.5379598140716553, | |
| "learning_rate": 3.904958677685951e-06, | |
| "loss": 0.5398801326751709, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.24813895781637718, | |
| "grad_norm": 3.455927848815918, | |
| "learning_rate": 4.111570247933885e-06, | |
| "loss": 0.5164627075195313, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.26054590570719605, | |
| "grad_norm": 3.3440027236938477, | |
| "learning_rate": 4.3181818181818185e-06, | |
| "loss": 0.5611258506774902, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.2729528535980149, | |
| "grad_norm": 3.941814661026001, | |
| "learning_rate": 4.524793388429753e-06, | |
| "loss": 0.5151082038879394, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.2853598014888337, | |
| "grad_norm": 3.3939054012298584, | |
| "learning_rate": 4.731404958677686e-06, | |
| "loss": 0.5357322216033935, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.2977667493796526, | |
| "grad_norm": 3.399260997772217, | |
| "learning_rate": 4.9380165289256205e-06, | |
| "loss": 0.5303369998931885, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.31017369727047145, | |
| "grad_norm": 3.109227180480957, | |
| "learning_rate": 4.999872331322402e-06, | |
| "loss": 0.49518985748291017, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.3225806451612903, | |
| "grad_norm": 3.147308588027954, | |
| "learning_rate": 4.999247046740511e-06, | |
| "loss": 0.505528736114502, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.3349875930521092, | |
| "grad_norm": 2.949301242828369, | |
| "learning_rate": 4.998100827074778e-06, | |
| "loss": 0.542401123046875, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.34739454094292804, | |
| "grad_norm": 3.650075912475586, | |
| "learning_rate": 4.996433911239573e-06, | |
| "loss": 0.511073637008667, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.3598014888337469, | |
| "grad_norm": 3.273531675338745, | |
| "learning_rate": 4.994246646681518e-06, | |
| "loss": 0.5162837982177735, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.37220843672456577, | |
| "grad_norm": 3.32936954498291, | |
| "learning_rate": 4.991539489307062e-06, | |
| "loss": 0.4840500831604004, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.38461538461538464, | |
| "grad_norm": 2.9847359657287598, | |
| "learning_rate": 4.988313003387454e-06, | |
| "loss": 0.48565964698791503, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.3970223325062035, | |
| "grad_norm": 3.2663910388946533, | |
| "learning_rate": 4.984567861441129e-06, | |
| "loss": 0.5017996788024902, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.4094292803970223, | |
| "grad_norm": 2.7149908542633057, | |
| "learning_rate": 4.980304844093531e-06, | |
| "loss": 0.4949507713317871, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.4218362282878412, | |
| "grad_norm": 3.2583069801330566, | |
| "learning_rate": 4.975524839914399e-06, | |
| "loss": 0.4920314311981201, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.43424317617866004, | |
| "grad_norm": 3.2360751628875732, | |
| "learning_rate": 4.970228845232561e-06, | |
| "loss": 0.4833060264587402, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.4466501240694789, | |
| "grad_norm": 3.0809836387634277, | |
| "learning_rate": 4.9644179639282595e-06, | |
| "loss": 0.4611049652099609, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.45905707196029777, | |
| "grad_norm": 2.903007984161377, | |
| "learning_rate": 4.958093407203059e-06, | |
| "loss": 0.4815668106079102, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.47146401985111663, | |
| "grad_norm": 3.1230084896087646, | |
| "learning_rate": 4.951256493327394e-06, | |
| "loss": 0.5097352981567382, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.4838709677419355, | |
| "grad_norm": 3.1644937992095947, | |
| "learning_rate": 4.943908647365789e-06, | |
| "loss": 0.46744871139526367, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.49627791563275436, | |
| "grad_norm": 3.0167317390441895, | |
| "learning_rate": 4.936051400879823e-06, | |
| "loss": 0.49645652770996096, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.5086848635235732, | |
| "grad_norm": 3.1851320266723633, | |
| "learning_rate": 4.9276863916088944e-06, | |
| "loss": 0.45388307571411135, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.5210918114143921, | |
| "grad_norm": 3.688347578048706, | |
| "learning_rate": 4.9188153631288584e-06, | |
| "loss": 0.48491249084472654, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.533498759305211, | |
| "grad_norm": 2.895963668823242, | |
| "learning_rate": 4.9094401644886e-06, | |
| "loss": 0.48116121292114256, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.5459057071960298, | |
| "grad_norm": 2.6067240238189697, | |
| "learning_rate": 4.899562749824624e-06, | |
| "loss": 0.45473780632019045, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.5583126550868487, | |
| "grad_norm": 2.9146952629089355, | |
| "learning_rate": 4.889185177953743e-06, | |
| "loss": 0.5124542236328125, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.5707196029776674, | |
| "grad_norm": 3.009894847869873, | |
| "learning_rate": 4.878309611943939e-06, | |
| "loss": 0.4542418956756592, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.5831265508684863, | |
| "grad_norm": 2.952033042907715, | |
| "learning_rate": 4.86693831866351e-06, | |
| "loss": 0.4757439136505127, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 0.5955334987593052, | |
| "grad_norm": 2.9270098209381104, | |
| "learning_rate": 4.855073668308561e-06, | |
| "loss": 0.48748092651367186, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 0.607940446650124, | |
| "grad_norm": 2.9587056636810303, | |
| "learning_rate": 4.842718133908973e-06, | |
| "loss": 0.48392572402954104, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 0.6203473945409429, | |
| "grad_norm": 2.895169734954834, | |
| "learning_rate": 4.829874290812932e-06, | |
| "loss": 0.49219765663146975, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.6327543424317618, | |
| "grad_norm": 2.6126275062561035, | |
| "learning_rate": 4.816544816150132e-06, | |
| "loss": 0.46747660636901855, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 0.6451612903225806, | |
| "grad_norm": 2.7628698348999023, | |
| "learning_rate": 4.802732488273762e-06, | |
| "loss": 0.4878196716308594, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 0.6575682382133995, | |
| "grad_norm": 2.8378942012786865, | |
| "learning_rate": 4.788440186181393e-06, | |
| "loss": 0.4586039066314697, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 0.6699751861042184, | |
| "grad_norm": 2.957749605178833, | |
| "learning_rate": 4.773670888914891e-06, | |
| "loss": 0.5078807353973389, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 0.6823821339950372, | |
| "grad_norm": 3.1145918369293213, | |
| "learning_rate": 4.75842767493948e-06, | |
| "loss": 0.4683982372283936, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.6947890818858561, | |
| "grad_norm": 2.674837827682495, | |
| "learning_rate": 4.742713721502065e-06, | |
| "loss": 0.45024881362915037, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 0.707196029776675, | |
| "grad_norm": 3.1005373001098633, | |
| "learning_rate": 4.726532303968986e-06, | |
| "loss": 0.45893230438232424, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 0.7196029776674938, | |
| "grad_norm": 2.7729763984680176, | |
| "learning_rate": 4.709886795143307e-06, | |
| "loss": 0.4667107582092285, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 0.7320099255583127, | |
| "grad_norm": 4.1112060546875, | |
| "learning_rate": 4.6927806645618034e-06, | |
| "loss": 0.43797926902770995, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 0.7444168734491315, | |
| "grad_norm": 3.006535291671753, | |
| "learning_rate": 4.675217477771779e-06, | |
| "loss": 0.4455252647399902, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.7568238213399504, | |
| "grad_norm": 2.7127737998962402, | |
| "learning_rate": 4.657200895587874e-06, | |
| "loss": 0.48641119003295896, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 0.7692307692307693, | |
| "grad_norm": 3.1489853858947754, | |
| "learning_rate": 4.638734673329023e-06, | |
| "loss": 0.4687007427215576, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 0.7816377171215881, | |
| "grad_norm": 2.6937718391418457, | |
| "learning_rate": 4.6198226600357e-06, | |
| "loss": 0.48453760147094727, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 0.794044665012407, | |
| "grad_norm": 3.03733229637146, | |
| "learning_rate": 4.600468797667641e-06, | |
| "loss": 0.44130840301513674, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 0.8064516129032258, | |
| "grad_norm": 2.9481587409973145, | |
| "learning_rate": 4.580677120282197e-06, | |
| "loss": 0.42711877822875977, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.8188585607940446, | |
| "grad_norm": 2.9542925357818604, | |
| "learning_rate": 4.56045175319348e-06, | |
| "loss": 0.4542987823486328, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 0.8312655086848635, | |
| "grad_norm": 2.777961492538452, | |
| "learning_rate": 4.539796912112505e-06, | |
| "loss": 0.42932806015014646, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 0.8436724565756824, | |
| "grad_norm": 3.0970194339752197, | |
| "learning_rate": 4.518716902268475e-06, | |
| "loss": 0.46704769134521484, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 0.8560794044665012, | |
| "grad_norm": 2.6771323680877686, | |
| "learning_rate": 4.497216117511413e-06, | |
| "loss": 0.4675713539123535, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 0.8684863523573201, | |
| "grad_norm": 3.0414323806762695, | |
| "learning_rate": 4.475299039396324e-06, | |
| "loss": 0.4250528335571289, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.8808933002481389, | |
| "grad_norm": 2.52348256111145, | |
| "learning_rate": 4.452970236249071e-06, | |
| "loss": 0.4666771411895752, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 0.8933002481389578, | |
| "grad_norm": 3.3117287158966064, | |
| "learning_rate": 4.4302343622141734e-06, | |
| "loss": 0.4492027282714844, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 0.9057071960297767, | |
| "grad_norm": 2.8878304958343506, | |
| "learning_rate": 4.4070961562847044e-06, | |
| "loss": 0.428986644744873, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 0.9181141439205955, | |
| "grad_norm": 2.7645702362060547, | |
| "learning_rate": 4.3835604413145175e-06, | |
| "loss": 0.4693061351776123, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 0.9305210918114144, | |
| "grad_norm": 2.759788751602173, | |
| "learning_rate": 4.3596321230129786e-06, | |
| "loss": 0.46426753997802733, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.9429280397022333, | |
| "grad_norm": 2.815812349319458, | |
| "learning_rate": 4.335316188922444e-06, | |
| "loss": 0.42464771270751955, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 0.9553349875930521, | |
| "grad_norm": 2.893314838409424, | |
| "learning_rate": 4.3106177073786684e-06, | |
| "loss": 0.4731471061706543, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 0.967741935483871, | |
| "grad_norm": 2.6007916927337646, | |
| "learning_rate": 4.285541826454374e-06, | |
| "loss": 0.43342103958129885, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 0.9801488833746899, | |
| "grad_norm": 2.937251329421997, | |
| "learning_rate": 4.26009377288621e-06, | |
| "loss": 0.4437389850616455, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 0.9925558312655087, | |
| "grad_norm": 2.6743578910827637, | |
| "learning_rate": 4.234278850985305e-06, | |
| "loss": 0.4493260860443115, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 1.0049627791563276, | |
| "grad_norm": 2.905144214630127, | |
| "learning_rate": 4.208102441531648e-06, | |
| "loss": 0.4088118553161621, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 1.0173697270471465, | |
| "grad_norm": 2.634845495223999, | |
| "learning_rate": 4.181570000652547e-06, | |
| "loss": 0.33449294567108157, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 1.0297766749379653, | |
| "grad_norm": 2.6692142486572266, | |
| "learning_rate": 4.15468705868536e-06, | |
| "loss": 0.3250420093536377, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 1.0421836228287842, | |
| "grad_norm": 2.6853787899017334, | |
| "learning_rate": 4.127459219024778e-06, | |
| "loss": 0.35315914154052735, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 1.054590570719603, | |
| "grad_norm": 3.406165361404419, | |
| "learning_rate": 4.099892156954861e-06, | |
| "loss": 0.3586670637130737, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 1.066997518610422, | |
| "grad_norm": 2.948725461959839, | |
| "learning_rate": 4.071991618466114e-06, | |
| "loss": 0.3647527933120728, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 1.0794044665012408, | |
| "grad_norm": 2.896860122680664, | |
| "learning_rate": 4.043763419057797e-06, | |
| "loss": 0.32327847480773925, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 1.0918114143920596, | |
| "grad_norm": 3.324622869491577, | |
| "learning_rate": 4.015213442525771e-06, | |
| "loss": 0.35637316703796384, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 1.1042183622828785, | |
| "grad_norm": 3.4257752895355225, | |
| "learning_rate": 3.986347639736092e-06, | |
| "loss": 0.33382978439331057, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 1.1166253101736974, | |
| "grad_norm": 3.2966017723083496, | |
| "learning_rate": 3.957172027384635e-06, | |
| "loss": 0.33941695690155027, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 1.129032258064516, | |
| "grad_norm": 3.0130667686462402, | |
| "learning_rate": 3.927692686742993e-06, | |
| "loss": 0.3642054319381714, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 1.141439205955335, | |
| "grad_norm": 3.649632453918457, | |
| "learning_rate": 3.897915762390912e-06, | |
| "loss": 0.3299143314361572, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 1.1538461538461537, | |
| "grad_norm": 3.1730785369873047, | |
| "learning_rate": 3.867847460935542e-06, | |
| "loss": 0.34182562828063967, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 1.1662531017369726, | |
| "grad_norm": 3.1270976066589355, | |
| "learning_rate": 3.8374940497177435e-06, | |
| "loss": 0.36127467155456544, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 1.1786600496277915, | |
| "grad_norm": 2.9266204833984375, | |
| "learning_rate": 3.806861855505748e-06, | |
| "loss": 0.3463557958602905, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 1.1910669975186103, | |
| "grad_norm": 3.1734328269958496, | |
| "learning_rate": 3.7759572631764243e-06, | |
| "loss": 0.34737136363983157, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 1.2034739454094292, | |
| "grad_norm": 3.387535333633423, | |
| "learning_rate": 3.7447867143844373e-06, | |
| "loss": 0.3464951038360596, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 1.215880893300248, | |
| "grad_norm": 3.2590081691741943, | |
| "learning_rate": 3.7133567062195697e-06, | |
| "loss": 0.3240300178527832, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 1.228287841191067, | |
| "grad_norm": 2.911224603652954, | |
| "learning_rate": 3.6816737898524844e-06, | |
| "loss": 0.33942902088165283, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 1.2406947890818858, | |
| "grad_norm": 3.3266444206237793, | |
| "learning_rate": 3.649744569169229e-06, | |
| "loss": 0.3586859226226807, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.2531017369727047, | |
| "grad_norm": 3.285763740539551, | |
| "learning_rate": 3.617575699394732e-06, | |
| "loss": 0.33882172107696534, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 1.2655086848635235, | |
| "grad_norm": 3.605529308319092, | |
| "learning_rate": 3.5851738857056174e-06, | |
| "loss": 0.34563610553741453, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 1.2779156327543424, | |
| "grad_norm": 2.4176876544952393, | |
| "learning_rate": 3.552545881832594e-06, | |
| "loss": 0.3259500741958618, | |
| "step": 1030 | |
| }, | |
| { | |
| "epoch": 1.2903225806451613, | |
| "grad_norm": 3.1849021911621094, | |
| "learning_rate": 3.519698488652734e-06, | |
| "loss": 0.35965259075164796, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 1.3027295285359801, | |
| "grad_norm": 3.008430004119873, | |
| "learning_rate": 3.486638552771914e-06, | |
| "loss": 0.34020233154296875, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 1.315136476426799, | |
| "grad_norm": 3.7227566242218018, | |
| "learning_rate": 3.4533729650977344e-06, | |
| "loss": 0.33164680004119873, | |
| "step": 1060 | |
| }, | |
| { | |
| "epoch": 1.3275434243176178, | |
| "grad_norm": 3.4713714122772217, | |
| "learning_rate": 3.4199086594031998e-06, | |
| "loss": 0.3493825435638428, | |
| "step": 1070 | |
| }, | |
| { | |
| "epoch": 1.3399503722084367, | |
| "grad_norm": 3.2516400814056396, | |
| "learning_rate": 3.386252610881465e-06, | |
| "loss": 0.3430775165557861, | |
| "step": 1080 | |
| }, | |
| { | |
| "epoch": 1.3523573200992556, | |
| "grad_norm": 3.756378650665283, | |
| "learning_rate": 3.3524118346919486e-06, | |
| "loss": 0.3269376277923584, | |
| "step": 1090 | |
| }, | |
| { | |
| "epoch": 1.3647642679900744, | |
| "grad_norm": 2.7790579795837402, | |
| "learning_rate": 3.318393384498114e-06, | |
| "loss": 0.3524430274963379, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.3771712158808933, | |
| "grad_norm": 3.254763603210449, | |
| "learning_rate": 3.2842043509972294e-06, | |
| "loss": 0.35284743309020994, | |
| "step": 1110 | |
| }, | |
| { | |
| "epoch": 1.3895781637717122, | |
| "grad_norm": 3.231228828430176, | |
| "learning_rate": 3.2498518604424005e-06, | |
| "loss": 0.35561535358428953, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 1.401985111662531, | |
| "grad_norm": 3.0170490741729736, | |
| "learning_rate": 3.215343073157204e-06, | |
| "loss": 0.36517820358276365, | |
| "step": 1130 | |
| }, | |
| { | |
| "epoch": 1.41439205955335, | |
| "grad_norm": 3.0740880966186523, | |
| "learning_rate": 3.180685182043205e-06, | |
| "loss": 0.34882569313049316, | |
| "step": 1140 | |
| }, | |
| { | |
| "epoch": 1.4267990074441688, | |
| "grad_norm": 3.487521171569824, | |
| "learning_rate": 3.1458854110806976e-06, | |
| "loss": 0.344951868057251, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 1.4392059553349876, | |
| "grad_norm": 2.7537150382995605, | |
| "learning_rate": 3.1109510138229565e-06, | |
| "loss": 0.3353364706039429, | |
| "step": 1160 | |
| }, | |
| { | |
| "epoch": 1.4516129032258065, | |
| "grad_norm": 2.8499064445495605, | |
| "learning_rate": 3.0758892718843352e-06, | |
| "loss": 0.33848204612731936, | |
| "step": 1170 | |
| }, | |
| { | |
| "epoch": 1.4640198511166254, | |
| "grad_norm": 3.2181148529052734, | |
| "learning_rate": 3.0407074934225038e-06, | |
| "loss": 0.3388556480407715, | |
| "step": 1180 | |
| }, | |
| { | |
| "epoch": 1.4764267990074442, | |
| "grad_norm": 3.099386215209961, | |
| "learning_rate": 3.005413011615165e-06, | |
| "loss": 0.3524498224258423, | |
| "step": 1190 | |
| }, | |
| { | |
| "epoch": 1.488833746898263, | |
| "grad_norm": 3.1054389476776123, | |
| "learning_rate": 2.9700131831315425e-06, | |
| "loss": 0.3347229719161987, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.501240694789082, | |
| "grad_norm": 2.9575774669647217, | |
| "learning_rate": 2.9345153865989918e-06, | |
| "loss": 0.321802020072937, | |
| "step": 1210 | |
| }, | |
| { | |
| "epoch": 1.5136476426799006, | |
| "grad_norm": 2.824366569519043, | |
| "learning_rate": 2.898927021065011e-06, | |
| "loss": 0.3323647499084473, | |
| "step": 1220 | |
| }, | |
| { | |
| "epoch": 1.5260545905707197, | |
| "grad_norm": 3.0914690494537354, | |
| "learning_rate": 2.8632555044550152e-06, | |
| "loss": 0.3432508945465088, | |
| "step": 1230 | |
| }, | |
| { | |
| "epoch": 1.5384615384615383, | |
| "grad_norm": 3.0662307739257812, | |
| "learning_rate": 2.8275082720261623e-06, | |
| "loss": 0.3417309522628784, | |
| "step": 1240 | |
| }, | |
| { | |
| "epoch": 1.5508684863523574, | |
| "grad_norm": 3.1436731815338135, | |
| "learning_rate": 2.791692774817576e-06, | |
| "loss": 0.3290053606033325, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 1.563275434243176, | |
| "grad_norm": 2.940727472305298, | |
| "learning_rate": 2.75581647809727e-06, | |
| "loss": 0.33008174896240233, | |
| "step": 1260 | |
| }, | |
| { | |
| "epoch": 1.5756823821339951, | |
| "grad_norm": 2.7560057640075684, | |
| "learning_rate": 2.7198868598061177e-06, | |
| "loss": 0.3164780616760254, | |
| "step": 1270 | |
| }, | |
| { | |
| "epoch": 1.5880893300248138, | |
| "grad_norm": 3.87307071685791, | |
| "learning_rate": 2.683911408999169e-06, | |
| "loss": 0.3081036567687988, | |
| "step": 1280 | |
| }, | |
| { | |
| "epoch": 1.6004962779156329, | |
| "grad_norm": 2.84149169921875, | |
| "learning_rate": 2.6478976242846576e-06, | |
| "loss": 0.35092926025390625, | |
| "step": 1290 | |
| }, | |
| { | |
| "epoch": 1.6129032258064515, | |
| "grad_norm": 3.040497303009033, | |
| "learning_rate": 2.6118530122610138e-06, | |
| "loss": 0.37850337028503417, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.6253101736972706, | |
| "grad_norm": 2.946133613586426, | |
| "learning_rate": 2.575785085952216e-06, | |
| "loss": 0.35753021240234373, | |
| "step": 1310 | |
| }, | |
| { | |
| "epoch": 1.6377171215880892, | |
| "grad_norm": 3.4700191020965576, | |
| "learning_rate": 2.539701363241795e-06, | |
| "loss": 0.34999761581420896, | |
| "step": 1320 | |
| }, | |
| { | |
| "epoch": 1.6501240694789083, | |
| "grad_norm": 2.9002044200897217, | |
| "learning_rate": 2.503609365305835e-06, | |
| "loss": 0.3304608345031738, | |
| "step": 1330 | |
| }, | |
| { | |
| "epoch": 1.662531017369727, | |
| "grad_norm": 3.0107545852661133, | |
| "learning_rate": 2.4675166150452814e-06, | |
| "loss": 0.3420397281646729, | |
| "step": 1340 | |
| }, | |
| { | |
| "epoch": 1.674937965260546, | |
| "grad_norm": 2.818993330001831, | |
| "learning_rate": 2.4314306355178924e-06, | |
| "loss": 0.31749625205993653, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 1.6873449131513647, | |
| "grad_norm": 2.8868680000305176, | |
| "learning_rate": 2.3953589483701556e-06, | |
| "loss": 0.33247082233428954, | |
| "step": 1360 | |
| }, | |
| { | |
| "epoch": 1.6997518610421838, | |
| "grad_norm": 2.5516133308410645, | |
| "learning_rate": 2.3593090722694995e-06, | |
| "loss": 0.3121840000152588, | |
| "step": 1370 | |
| }, | |
| { | |
| "epoch": 1.7121588089330024, | |
| "grad_norm": 2.8889875411987305, | |
| "learning_rate": 2.3232885213371297e-06, | |
| "loss": 0.3375117778778076, | |
| "step": 1380 | |
| }, | |
| { | |
| "epoch": 1.7245657568238213, | |
| "grad_norm": 3.4715304374694824, | |
| "learning_rate": 2.2873048035818043e-06, | |
| "loss": 0.3526495695114136, | |
| "step": 1390 | |
| }, | |
| { | |
| "epoch": 1.7369727047146402, | |
| "grad_norm": 2.6388895511627197, | |
| "learning_rate": 2.251365419334895e-06, | |
| "loss": 0.3385695219039917, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.749379652605459, | |
| "grad_norm": 3.295314311981201, | |
| "learning_rate": 2.2154778596870365e-06, | |
| "loss": 0.3281654119491577, | |
| "step": 1410 | |
| }, | |
| { | |
| "epoch": 1.7617866004962779, | |
| "grad_norm": 2.9345321655273438, | |
| "learning_rate": 2.179649604926708e-06, | |
| "loss": 0.3286000728607178, | |
| "step": 1420 | |
| }, | |
| { | |
| "epoch": 1.7741935483870968, | |
| "grad_norm": 3.245892286300659, | |
| "learning_rate": 2.14388812298107e-06, | |
| "loss": 0.31292095184326174, | |
| "step": 1430 | |
| }, | |
| { | |
| "epoch": 1.7866004962779156, | |
| "grad_norm": 2.8466174602508545, | |
| "learning_rate": 2.108200867859369e-06, | |
| "loss": 0.32442235946655273, | |
| "step": 1440 | |
| }, | |
| { | |
| "epoch": 1.7990074441687345, | |
| "grad_norm": 3.2279984951019287, | |
| "learning_rate": 2.072595278099247e-06, | |
| "loss": 0.3294026851654053, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 1.8114143920595533, | |
| "grad_norm": 3.1583516597747803, | |
| "learning_rate": 2.037078775216283e-06, | |
| "loss": 0.3263838768005371, | |
| "step": 1460 | |
| }, | |
| { | |
| "epoch": 1.8238213399503722, | |
| "grad_norm": 3.04581356048584, | |
| "learning_rate": 2.0016587621570682e-06, | |
| "loss": 0.33864445686340333, | |
| "step": 1470 | |
| }, | |
| { | |
| "epoch": 1.836228287841191, | |
| "grad_norm": 3.7386040687561035, | |
| "learning_rate": 1.966342621756163e-06, | |
| "loss": 0.31294050216674807, | |
| "step": 1480 | |
| }, | |
| { | |
| "epoch": 1.84863523573201, | |
| "grad_norm": 2.868602991104126, | |
| "learning_rate": 1.931137715197242e-06, | |
| "loss": 0.31308484077453613, | |
| "step": 1490 | |
| }, | |
| { | |
| "epoch": 1.8610421836228288, | |
| "grad_norm": 3.055979013442993, | |
| "learning_rate": 1.8960513804787557e-06, | |
| "loss": 0.333689284324646, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.8734491315136477, | |
| "grad_norm": 3.1473002433776855, | |
| "learning_rate": 1.861090930884412e-06, | |
| "loss": 0.33293652534484863, | |
| "step": 1510 | |
| }, | |
| { | |
| "epoch": 1.8858560794044665, | |
| "grad_norm": 2.9485549926757812, | |
| "learning_rate": 1.8262636534588316e-06, | |
| "loss": 0.3332956790924072, | |
| "step": 1520 | |
| }, | |
| { | |
| "epoch": 1.8982630272952854, | |
| "grad_norm": 2.9611082077026367, | |
| "learning_rate": 1.7915768074886497e-06, | |
| "loss": 0.32119269371032716, | |
| "step": 1530 | |
| }, | |
| { | |
| "epoch": 1.9106699751861043, | |
| "grad_norm": 2.7273433208465576, | |
| "learning_rate": 1.7570376229894164e-06, | |
| "loss": 0.32526330947875975, | |
| "step": 1540 | |
| }, | |
| { | |
| "epoch": 1.9230769230769231, | |
| "grad_norm": 3.0661497116088867, | |
| "learning_rate": 1.7226532991985976e-06, | |
| "loss": 0.3218840599060059, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 1.935483870967742, | |
| "grad_norm": 2.9829108715057373, | |
| "learning_rate": 1.688431003074985e-06, | |
| "loss": 0.3307999610900879, | |
| "step": 1560 | |
| }, | |
| { | |
| "epoch": 1.9478908188585606, | |
| "grad_norm": 3.018162488937378, | |
| "learning_rate": 1.654377867804841e-06, | |
| "loss": 0.34044556617736815, | |
| "step": 1570 | |
| }, | |
| { | |
| "epoch": 1.9602977667493797, | |
| "grad_norm": 2.9301137924194336, | |
| "learning_rate": 1.6205009913150815e-06, | |
| "loss": 0.31292636394500734, | |
| "step": 1580 | |
| }, | |
| { | |
| "epoch": 1.9727047146401984, | |
| "grad_norm": 2.9100582599639893, | |
| "learning_rate": 1.5868074347937993e-06, | |
| "loss": 0.32329092025756834, | |
| "step": 1590 | |
| }, | |
| { | |
| "epoch": 1.9851116625310175, | |
| "grad_norm": 2.979872465133667, | |
| "learning_rate": 1.553304221218462e-06, | |
| "loss": 0.3566758155822754, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 1.997518610421836, | |
| "grad_norm": 2.6494853496551514, | |
| "learning_rate": 1.5199983338920594e-06, | |
| "loss": 0.3069176197052002, | |
| "step": 1610 | |
| }, | |
| { | |
| "epoch": 2.009925558312655, | |
| "grad_norm": 2.3948535919189453, | |
| "learning_rate": 1.4868967149875257e-06, | |
| "loss": 0.25013885498046873, | |
| "step": 1620 | |
| }, | |
| { | |
| "epoch": 2.022332506203474, | |
| "grad_norm": 2.941615343093872, | |
| "learning_rate": 1.4540062641007383e-06, | |
| "loss": 0.22131125926971434, | |
| "step": 1630 | |
| }, | |
| { | |
| "epoch": 2.034739454094293, | |
| "grad_norm": 3.205547332763672, | |
| "learning_rate": 1.421333836812387e-06, | |
| "loss": 0.2237858295440674, | |
| "step": 1640 | |
| }, | |
| { | |
| "epoch": 2.0471464019851116, | |
| "grad_norm": 3.0743908882141113, | |
| "learning_rate": 1.3888862432590167e-06, | |
| "loss": 0.22658278942108154, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 2.0595533498759306, | |
| "grad_norm": 2.8300976753234863, | |
| "learning_rate": 1.3566702467135445e-06, | |
| "loss": 0.2289585828781128, | |
| "step": 1660 | |
| }, | |
| { | |
| "epoch": 2.0719602977667493, | |
| "grad_norm": 2.5701904296875, | |
| "learning_rate": 1.3246925621755496e-06, | |
| "loss": 0.22105908393859863, | |
| "step": 1670 | |
| }, | |
| { | |
| "epoch": 2.0843672456575684, | |
| "grad_norm": 2.67372727394104, | |
| "learning_rate": 1.2929598549716122e-06, | |
| "loss": 0.20758159160614015, | |
| "step": 1680 | |
| }, | |
| { | |
| "epoch": 2.096774193548387, | |
| "grad_norm": 3.0295441150665283, | |
| "learning_rate": 1.2614787393660215e-06, | |
| "loss": 0.2294921875, | |
| "step": 1690 | |
| }, | |
| { | |
| "epoch": 2.109181141439206, | |
| "grad_norm": 3.321169137954712, | |
| "learning_rate": 1.2302557771821154e-06, | |
| "loss": 0.22693567276000975, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 2.1215880893300247, | |
| "grad_norm": 3.0045440196990967, | |
| "learning_rate": 1.199297476434553e-06, | |
| "loss": 0.23366827964782716, | |
| "step": 1710 | |
| }, | |
| { | |
| "epoch": 2.133995037220844, | |
| "grad_norm": 3.3027615547180176, | |
| "learning_rate": 1.168610289972813e-06, | |
| "loss": 0.204612398147583, | |
| "step": 1720 | |
| }, | |
| { | |
| "epoch": 2.1464019851116625, | |
| "grad_norm": 3.4568586349487305, | |
| "learning_rate": 1.138200614136172e-06, | |
| "loss": 0.23412892818450928, | |
| "step": 1730 | |
| }, | |
| { | |
| "epoch": 2.1588089330024816, | |
| "grad_norm": 3.1891727447509766, | |
| "learning_rate": 1.1080747874204809e-06, | |
| "loss": 0.23051118850708008, | |
| "step": 1740 | |
| }, | |
| { | |
| "epoch": 2.1712158808933, | |
| "grad_norm": 2.898972511291504, | |
| "learning_rate": 1.0782390891569822e-06, | |
| "loss": 0.223101806640625, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 2.1836228287841193, | |
| "grad_norm": 3.198352098464966, | |
| "learning_rate": 1.0486997382034695e-06, | |
| "loss": 0.21067981719970702, | |
| "step": 1760 | |
| }, | |
| { | |
| "epoch": 2.196029776674938, | |
| "grad_norm": 3.239396095275879, | |
| "learning_rate": 1.0194628916480504e-06, | |
| "loss": 0.21455581188201905, | |
| "step": 1770 | |
| }, | |
| { | |
| "epoch": 2.208436724565757, | |
| "grad_norm": 3.6431620121002197, | |
| "learning_rate": 9.905346435257784e-07, | |
| "loss": 0.25631544589996336, | |
| "step": 1780 | |
| }, | |
| { | |
| "epoch": 2.2208436724565757, | |
| "grad_norm": 3.2421746253967285, | |
| "learning_rate": 9.619210235484333e-07, | |
| "loss": 0.2324852466583252, | |
| "step": 1790 | |
| }, | |
| { | |
| "epoch": 2.2332506203473947, | |
| "grad_norm": 3.0301804542541504, | |
| "learning_rate": 9.336279958477096e-07, | |
| "loss": 0.2013075351715088, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 2.2456575682382134, | |
| "grad_norm": 3.0917258262634277, | |
| "learning_rate": 9.056614577320702e-07, | |
| "loss": 0.23809702396392823, | |
| "step": 1810 | |
| }, | |
| { | |
| "epoch": 2.258064516129032, | |
| "grad_norm": 3.2016923427581787, | |
| "learning_rate": 8.780272384575289e-07, | |
| "loss": 0.22165682315826415, | |
| "step": 1820 | |
| }, | |
| { | |
| "epoch": 2.270471464019851, | |
| "grad_norm": 3.2964746952056885, | |
| "learning_rate": 8.507310980126188e-07, | |
| "loss": 0.21906075477600098, | |
| "step": 1830 | |
| }, | |
| { | |
| "epoch": 2.28287841191067, | |
| "grad_norm": 3.021667242050171, | |
| "learning_rate": 8.237787259178037e-07, | |
| "loss": 0.21673779487609862, | |
| "step": 1840 | |
| }, | |
| { | |
| "epoch": 2.295285359801489, | |
| "grad_norm": 3.4305264949798584, | |
| "learning_rate": 7.971757400395652e-07, | |
| "loss": 0.23625686168670654, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 2.3076923076923075, | |
| "grad_norm": 2.7715115547180176, | |
| "learning_rate": 7.709276854194392e-07, | |
| "loss": 0.21259665489196777, | |
| "step": 1860 | |
| }, | |
| { | |
| "epoch": 2.3200992555831266, | |
| "grad_norm": 3.520462989807129, | |
| "learning_rate": 7.450400331182198e-07, | |
| "loss": 0.23154499530792236, | |
| "step": 1870 | |
| }, | |
| { | |
| "epoch": 2.332506203473945, | |
| "grad_norm": 3.7826356887817383, | |
| "learning_rate": 7.19518179075591e-07, | |
| "loss": 0.2277855396270752, | |
| "step": 1880 | |
| }, | |
| { | |
| "epoch": 2.3449131513647643, | |
| "grad_norm": 3.4630792140960693, | |
| "learning_rate": 6.943674429854172e-07, | |
| "loss": 0.22485933303833008, | |
| "step": 1890 | |
| }, | |
| { | |
| "epoch": 2.357320099255583, | |
| "grad_norm": 3.305630922317505, | |
| "learning_rate": 6.695930671869175e-07, | |
| "loss": 0.23359484672546388, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 2.369727047146402, | |
| "grad_norm": 2.659604787826538, | |
| "learning_rate": 6.452002155719725e-07, | |
| "loss": 0.19909019470214845, | |
| "step": 1910 | |
| }, | |
| { | |
| "epoch": 2.3821339950372207, | |
| "grad_norm": 3.414795398712158, | |
| "learning_rate": 6.211939725087806e-07, | |
| "loss": 0.23796792030334474, | |
| "step": 1920 | |
| }, | |
| { | |
| "epoch": 2.3945409429280398, | |
| "grad_norm": 3.0635950565338135, | |
| "learning_rate": 5.975793417820841e-07, | |
| "loss": 0.22071402072906493, | |
| "step": 1930 | |
| }, | |
| { | |
| "epoch": 2.4069478908188584, | |
| "grad_norm": 3.123188018798828, | |
| "learning_rate": 5.743612455502016e-07, | |
| "loss": 0.22007994651794432, | |
| "step": 1940 | |
| }, | |
| { | |
| "epoch": 2.4193548387096775, | |
| "grad_norm": 2.785230875015259, | |
| "learning_rate": 5.515445233190636e-07, | |
| "loss": 0.21707756519317628, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 2.431761786600496, | |
| "grad_norm": 3.8641016483306885, | |
| "learning_rate": 5.291339309334845e-07, | |
| "loss": 0.2090601682662964, | |
| "step": 1960 | |
| }, | |
| { | |
| "epoch": 2.444168734491315, | |
| "grad_norm": 2.880202054977417, | |
| "learning_rate": 5.071341395858703e-07, | |
| "loss": 0.2203066825866699, | |
| "step": 1970 | |
| }, | |
| { | |
| "epoch": 2.456575682382134, | |
| "grad_norm": 3.224031448364258, | |
| "learning_rate": 4.855497348425686e-07, | |
| "loss": 0.22847747802734375, | |
| "step": 1980 | |
| }, | |
| { | |
| "epoch": 2.468982630272953, | |
| "grad_norm": 3.334984302520752, | |
| "learning_rate": 4.6438521568806625e-07, | |
| "loss": 0.20745439529418946, | |
| "step": 1990 | |
| }, | |
| { | |
| "epoch": 2.4813895781637716, | |
| "grad_norm": 3.121742010116577, | |
| "learning_rate": 4.436449935872378e-07, | |
| "loss": 0.23231451511383056, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.4937965260545907, | |
| "grad_norm": 2.765779972076416, | |
| "learning_rate": 4.2333339156583467e-07, | |
| "loss": 0.2055814266204834, | |
| "step": 2010 | |
| }, | |
| { | |
| "epoch": 2.5062034739454093, | |
| "grad_norm": 3.019102096557617, | |
| "learning_rate": 4.034546433094036e-07, | |
| "loss": 0.22689828872680665, | |
| "step": 2020 | |
| }, | |
| { | |
| "epoch": 2.5186104218362284, | |
| "grad_norm": 3.2937521934509277, | |
| "learning_rate": 3.8401289228083395e-07, | |
| "loss": 0.23529858589172364, | |
| "step": 2030 | |
| }, | |
| { | |
| "epoch": 2.531017369727047, | |
| "grad_norm": 3.353877544403076, | |
| "learning_rate": 3.6501219085670724e-07, | |
| "loss": 0.2225353479385376, | |
| "step": 2040 | |
| }, | |
| { | |
| "epoch": 2.543424317617866, | |
| "grad_norm": 3.6522250175476074, | |
| "learning_rate": 3.4645649948263085e-07, | |
| "loss": 0.2311680793762207, | |
| "step": 2050 | |
| }, | |
| { | |
| "epoch": 2.555831265508685, | |
| "grad_norm": 3.3297574520111084, | |
| "learning_rate": 3.2834968584773754e-07, | |
| "loss": 0.21333532333374022, | |
| "step": 2060 | |
| }, | |
| { | |
| "epoch": 2.568238213399504, | |
| "grad_norm": 3.3023574352264404, | |
| "learning_rate": 3.106955240785153e-07, | |
| "loss": 0.2413243293762207, | |
| "step": 2070 | |
| }, | |
| { | |
| "epoch": 2.5806451612903225, | |
| "grad_norm": 2.7704153060913086, | |
| "learning_rate": 2.9349769395214e-07, | |
| "loss": 0.21133730411529542, | |
| "step": 2080 | |
| }, | |
| { | |
| "epoch": 2.5930521091811416, | |
| "grad_norm": 3.1937754154205322, | |
| "learning_rate": 2.767597801294772e-07, | |
| "loss": 0.21675755977630615, | |
| "step": 2090 | |
| }, | |
| { | |
| "epoch": 2.6054590570719602, | |
| "grad_norm": 3.397535562515259, | |
| "learning_rate": 2.604852714079023e-07, | |
| "loss": 0.22542505264282225, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.617866004962779, | |
| "grad_norm": 3.3038713932037354, | |
| "learning_rate": 2.4467755999411204e-07, | |
| "loss": 0.23392457962036134, | |
| "step": 2110 | |
| }, | |
| { | |
| "epoch": 2.630272952853598, | |
| "grad_norm": 4.148429870605469, | |
| "learning_rate": 2.2933994079705896e-07, | |
| "loss": 0.19952282905578614, | |
| "step": 2120 | |
| }, | |
| { | |
| "epoch": 2.642679900744417, | |
| "grad_norm": 3.0992136001586914, | |
| "learning_rate": 2.144756107411733e-07, | |
| "loss": 0.2230466365814209, | |
| "step": 2130 | |
| }, | |
| { | |
| "epoch": 2.6550868486352357, | |
| "grad_norm": 3.451746940612793, | |
| "learning_rate": 2.0008766810000801e-07, | |
| "loss": 0.23057200908660888, | |
| "step": 2140 | |
| }, | |
| { | |
| "epoch": 2.6674937965260543, | |
| "grad_norm": 3.058110237121582, | |
| "learning_rate": 1.861791118504408e-07, | |
| "loss": 0.2180851459503174, | |
| "step": 2150 | |
| }, | |
| { | |
| "epoch": 2.6799007444168734, | |
| "grad_norm": 2.745176315307617, | |
| "learning_rate": 1.7275284104757844e-07, | |
| "loss": 0.2167917251586914, | |
| "step": 2160 | |
| }, | |
| { | |
| "epoch": 2.6923076923076925, | |
| "grad_norm": 3.708021402359009, | |
| "learning_rate": 1.598116542204889e-07, | |
| "loss": 0.22899148464202881, | |
| "step": 2170 | |
| }, | |
| { | |
| "epoch": 2.704714640198511, | |
| "grad_norm": 3.6781704425811768, | |
| "learning_rate": 1.4735824878888277e-07, | |
| "loss": 0.2170346736907959, | |
| "step": 2180 | |
| }, | |
| { | |
| "epoch": 2.71712158808933, | |
| "grad_norm": 2.478654623031616, | |
| "learning_rate": 1.353952205008713e-07, | |
| "loss": 0.2242295265197754, | |
| "step": 2190 | |
| }, | |
| { | |
| "epoch": 2.729528535980149, | |
| "grad_norm": 2.7087976932525635, | |
| "learning_rate": 1.2392506289191837e-07, | |
| "loss": 0.2131351947784424, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.741935483870968, | |
| "grad_norm": 2.867835283279419, | |
| "learning_rate": 1.1295016676509618e-07, | |
| "loss": 0.21861982345581055, | |
| "step": 2210 | |
| }, | |
| { | |
| "epoch": 2.7543424317617866, | |
| "grad_norm": 2.9449636936187744, | |
| "learning_rate": 1.0247281969275131e-07, | |
| "loss": 0.1985526442527771, | |
| "step": 2220 | |
| }, | |
| { | |
| "epoch": 2.7667493796526053, | |
| "grad_norm": 3.0885448455810547, | |
| "learning_rate": 9.249520553969465e-08, | |
| "loss": 0.23580389022827147, | |
| "step": 2230 | |
| }, | |
| { | |
| "epoch": 2.7791563275434243, | |
| "grad_norm": 3.1814591884613037, | |
| "learning_rate": 8.301940400799818e-08, | |
| "loss": 0.21973462104797364, | |
| "step": 2240 | |
| }, | |
| { | |
| "epoch": 2.7915632754342434, | |
| "grad_norm": 3.26047682762146, | |
| "learning_rate": 7.404739020351381e-08, | |
| "loss": 0.22958881855010987, | |
| "step": 2250 | |
| }, | |
| { | |
| "epoch": 2.803970223325062, | |
| "grad_norm": 3.2021138668060303, | |
| "learning_rate": 6.558103422418555e-08, | |
| "loss": 0.23037824630737305, | |
| "step": 2260 | |
| }, | |
| { | |
| "epoch": 2.8163771712158807, | |
| "grad_norm": 3.562350273132324, | |
| "learning_rate": 5.762210077025293e-08, | |
| "loss": 0.2314563512802124, | |
| "step": 2270 | |
| }, | |
| { | |
| "epoch": 2.8287841191067, | |
| "grad_norm": 3.15472412109375, | |
| "learning_rate": 5.017224877642196e-08, | |
| "loss": 0.21394333839416504, | |
| "step": 2280 | |
| }, | |
| { | |
| "epoch": 2.8411910669975184, | |
| "grad_norm": 2.9189085960388184, | |
| "learning_rate": 4.323303106608362e-08, | |
| "loss": 0.22109608650207518, | |
| "step": 2290 | |
| }, | |
| { | |
| "epoch": 2.8535980148883375, | |
| "grad_norm": 3.364637851715088, | |
| "learning_rate": 3.680589402764695e-08, | |
| "loss": 0.21065773963928222, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.866004962779156, | |
| "grad_norm": 3.1239407062530518, | |
| "learning_rate": 3.089217731305882e-08, | |
| "loss": 0.21384596824645996, | |
| "step": 2310 | |
| }, | |
| { | |
| "epoch": 2.8784119106699753, | |
| "grad_norm": 3.2410709857940674, | |
| "learning_rate": 2.5493113558571203e-08, | |
| "loss": 0.2103095531463623, | |
| "step": 2320 | |
| }, | |
| { | |
| "epoch": 2.890818858560794, | |
| "grad_norm": 3.29823899269104, | |
| "learning_rate": 2.0609828127814425e-08, | |
| "loss": 0.22623074054718018, | |
| "step": 2330 | |
| }, | |
| { | |
| "epoch": 2.903225806451613, | |
| "grad_norm": 3.447692632675171, | |
| "learning_rate": 1.624333887723012e-08, | |
| "loss": 0.2561406373977661, | |
| "step": 2340 | |
| }, | |
| { | |
| "epoch": 2.9156327543424316, | |
| "grad_norm": 3.5823864936828613, | |
| "learning_rate": 1.2394555943911757e-08, | |
| "loss": 0.22645580768585205, | |
| "step": 2350 | |
| }, | |
| { | |
| "epoch": 2.9280397022332507, | |
| "grad_norm": 3.032896041870117, | |
| "learning_rate": 9.0642815559e-09, | |
| "loss": 0.21149790287017822, | |
| "step": 2360 | |
| }, | |
| { | |
| "epoch": 2.9404466501240694, | |
| "grad_norm": 3.0762665271759033, | |
| "learning_rate": 6.253209864966747e-09, | |
| "loss": 0.2031836986541748, | |
| "step": 2370 | |
| }, | |
| { | |
| "epoch": 2.9528535980148884, | |
| "grad_norm": 3.5435872077941895, | |
| "learning_rate": 3.9619268019305825e-09, | |
| "loss": 0.2176318883895874, | |
| "step": 2380 | |
| }, | |
| { | |
| "epoch": 2.965260545905707, | |
| "grad_norm": 3.506044387817383, | |
| "learning_rate": 2.190909954525866e-09, | |
| "loss": 0.2153688907623291, | |
| "step": 2390 | |
| }, | |
| { | |
| "epoch": 2.977667493796526, | |
| "grad_norm": 3.4698026180267334, | |
| "learning_rate": 9.405284678556948e-10, | |
| "loss": 0.21965882778167725, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 2.990074441687345, | |
| "grad_norm": 3.0432729721069336, | |
| "learning_rate": 2.1104296745011466e-10, | |
| "loss": 0.2195052146911621, | |
| "step": 2410 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "step": 2418, | |
| "total_flos": 1.329254778357678e+18, | |
| "train_loss": 0.361527387419449, | |
| "train_runtime": 121252.8356, | |
| "train_samples_per_second": 0.16, | |
| "train_steps_per_second": 0.02 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 2418, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 200, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.329254778357678e+18, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |