| [2025-07-05 14:38:41,543][__main__][INFO] - cache_dir: /tmp/ |
| dataset: |
| name: kamel-usp/aes_enem_dataset |
| split: JBCS2025 |
| training_params: |
| seed: 42 |
| num_train_epochs: 20 |
| logging_steps: 100 |
| metric_for_best_model: QWK |
| bf16: true |
| bootstrap: |
| enabled: true |
| n_bootstrap: 10000 |
| bootstrap_seed: 42 |
| metrics: |
| - QWK |
| - Macro_F1 |
| - Weighted_F1 |
| post_training_results: |
| model_path: /workspace/jbcs2025/outputs/2025-03-24/20-42-59 |
| experiments: |
| model: |
| name: meta-llama/Llama-3.1-8B |
| type: llama31_classification_lora |
| num_labels: 6 |
| output_dir: ./results/ |
| logging_dir: ./logs/ |
| best_model_dir: ./results/best_model |
| lora_r: 16 |
| lora_dropout: 0.1 |
| lora_alpha: 32 |
| lora_target_modules: all-linear |
| tokenizer: |
| name: meta-llama/Llama-3.1-8B |
| dataset: |
| grade_index: 0 |
| use_full_context: true |
| training_params: |
| weight_decay: 0.01 |
| warmup_ratio: 0.1 |
| learning_rate: 5.0e-05 |
| train_batch_size: 8 |
| eval_batch_size: 4 |
| gradient_accumulation_steps: 2 |
| gradient_checkpointing: true |
|
|
| [2025-07-05 14:38:45,473][__main__][INFO] - GPU 0: NVIDIA H200 | TDP ≈ 700 W |
| [2025-07-05 14:38:45,473][__main__][INFO] - Starting the Fine Tuning training process. |
| [2025-07-05 14:38:49,913][transformers.tokenization_utils_base][INFO] - loading file tokenizer.json from cache at /tmp/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/tokenizer.json |
| [2025-07-05 14:38:49,914][transformers.tokenization_utils_base][INFO] - loading file tokenizer.model from cache at None |
| [2025-07-05 14:38:49,914][transformers.tokenization_utils_base][INFO] - loading file added_tokens.json from cache at None |
| [2025-07-05 14:38:49,914][transformers.tokenization_utils_base][INFO] - loading file special_tokens_map.json from cache at /tmp/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/special_tokens_map.json |
| [2025-07-05 14:38:49,914][transformers.tokenization_utils_base][INFO] - loading file tokenizer_config.json from cache at /tmp/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/tokenizer_config.json |
| [2025-07-05 14:38:49,914][transformers.tokenization_utils_base][INFO] - loading file chat_template.jinja from cache at None |
| [2025-07-05 14:38:50,189][transformers.tokenization_utils_base][INFO] - Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained. |
| [2025-07-05 14:38:50,194][__main__][INFO] - Tokenizer function parameters- Padding:longest; Truncation: False; Use Full Context: True |
| [2025-07-05 14:38:50,806][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /tmp/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 14:38:50,808][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "id2label": { |
| "0": "LABEL_0", |
| "1": "LABEL_1", |
| "2": "LABEL_2", |
| "3": "LABEL_3", |
| "4": "LABEL_4", |
| "5": "LABEL_5" |
| }, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "label2id": { |
| "LABEL_0": 0, |
| "LABEL_1": 1, |
| "LABEL_2": 2, |
| "LABEL_3": 3, |
| "LABEL_4": 4, |
| "LABEL_5": 5 |
| }, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 14:38:50,940][transformers.modeling_utils][INFO] - loading weights file model.safetensors from cache at /tmp/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/model.safetensors.index.json |
| [2025-07-05 14:38:50,940][transformers.modeling_utils][INFO] - Will use torch_dtype=torch.bfloat16 as defined in model's config object |
| [2025-07-05 14:38:50,940][transformers.modeling_utils][INFO] - Instantiating LlamaForSequenceClassification model under default dtype torch.bfloat16. |
| [2025-07-05 14:39:03,780][transformers.modeling_utils][INFO] - Some weights of the model checkpoint at meta-llama/Llama-3.1-8B were not used when initializing LlamaForSequenceClassification: ['lm_head.weight'] |
| - This IS expected if you are initializing LlamaForSequenceClassification from the checkpoint of a model trained on another task or with another architecture (e.g. initializing a BertForSequenceClassification model from a BertForPreTraining model). |
| - This IS NOT expected if you are initializing LlamaForSequenceClassification from the checkpoint of a model that you expect to be exactly identical (initializing a BertForSequenceClassification model from a BertForSequenceClassification model). |
| [2025-07-05 14:39:03,781][transformers.modeling_utils][WARNING] - Some weights of LlamaForSequenceClassification were not initialized from the model checkpoint at meta-llama/Llama-3.1-8B and are newly initialized: ['score.weight'] |
| You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference. |
| [2025-07-05 14:39:05,986][__main__][INFO] - Initialized new PEFT model for ce loss |
| [2025-07-05 14:39:05,989][__main__][INFO] - None |
| [2025-07-05 14:39:05,990][transformers.training_args][INFO] - PyTorch: setting up devices |
| [2025-07-05 14:39:06,027][__main__][INFO] - Total steps: 620. Number of warmup steps: 62 |
| [2025-07-05 14:39:06,033][transformers.trainer][INFO] - You have loaded a model on multiple GPUs. `is_model_parallel` attribute will be force-set to `True` to avoid any unexpected behavior such as device placement mismatching. |
| [2025-07-05 14:39:06,058][transformers.trainer][INFO] - Using auto half precision backend |
| [2025-07-05 14:39:06,059][transformers.trainer][WARNING] - No label_names provided for model class `PeftModelForSequenceClassification`. Since `PeftModel` hides base models input arguments, if label_names is not given, label_names can't be set automatically within `Trainer`. Note that empty label_names list will be used instead. |
| [2025-07-05 14:39:06,060][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 14:39:06,076][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 14:39:06,076][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 14:39:06,076][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 14:39:27,512][transformers.trainer][INFO] - The following columns in the Training set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - ***** Running training ***** |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - Num examples = 500 |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - Num Epochs = 20 |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - Instantaneous batch size per device = 8 |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - Total train batch size (w. parallel, distributed & accumulation) = 16 |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - Gradient Accumulation steps = 2 |
| [2025-07-05 14:39:27,565][transformers.trainer][INFO] - Total optimization steps = 640 |
| [2025-07-05 14:39:27,568][transformers.trainer][INFO] - Number of trainable parameters = 41,967,616 |
| [2025-07-05 14:39:27,647][transformers.models.llama.modeling_llama][WARNING] - `use_cache=True` is incompatible with gradient checkpointing. Setting `use_cache=False`. |
| [2025-07-05 14:43:53,780][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 14:43:53,784][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 14:43:53,784][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 14:43:53,784][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 14:44:14,825][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-32 |
| [2025-07-05 14:44:15,173][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 14:44:15,174][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 14:48:42,177][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 14:48:42,181][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 14:48:42,181][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 14:48:42,181][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 14:49:03,226][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-64 |
| [2025-07-05 14:49:03,578][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 14:49:03,578][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 14:53:30,388][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 14:53:30,392][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 14:53:30,392][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 14:53:30,392][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 14:53:51,427][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-96 |
| [2025-07-05 14:53:51,753][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 14:53:51,753][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 14:53:52,501][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-32] due to args.save_total_limit |
| [2025-07-05 14:53:52,529][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-64] due to args.save_total_limit |
| [2025-07-05 14:58:18,376][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 14:58:18,380][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 14:58:18,380][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 14:58:18,380][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 14:58:39,410][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-128 |
| [2025-07-05 14:58:39,898][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 14:58:39,899][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 14:58:40,579][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-96] due to args.save_total_limit |
| [2025-07-05 15:03:06,525][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:03:06,529][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:03:06,529][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:03:06,529][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:03:27,571][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-160 |
| [2025-07-05 15:03:27,894][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:03:27,895][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:03:28,588][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-128] due to args.save_total_limit |
| [2025-07-05 15:07:54,642][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:07:54,647][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:07:54,647][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:07:54,647][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:08:15,673][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-192 |
| [2025-07-05 15:08:16,010][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:08:16,011][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:08:16,774][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-160] due to args.save_total_limit |
| [2025-07-05 15:12:42,833][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:12:42,837][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:12:42,837][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:12:42,837][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:13:03,876][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-224 |
| [2025-07-05 15:13:04,202][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:13:04,202][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:13:04,895][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-192] due to args.save_total_limit |
| [2025-07-05 15:17:30,766][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:17:30,770][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:17:30,770][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:17:30,770][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:17:51,838][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-256 |
| [2025-07-05 15:17:52,160][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:17:52,161][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:22:18,671][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:22:18,675][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:22:18,675][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:22:18,675][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:22:39,699][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-288 |
| [2025-07-05 15:22:40,024][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:22:40,025][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:22:40,789][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-256] due to args.save_total_limit |
| [2025-07-05 15:27:06,639][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:27:06,643][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:27:06,643][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:27:06,643][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:27:27,681][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-320 |
| [2025-07-05 15:27:28,010][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:27:28,011][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:27:28,775][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-288] due to args.save_total_limit |
| [2025-07-05 15:31:54,723][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:31:54,727][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:31:54,727][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:31:54,727][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:32:15,764][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-352 |
| [2025-07-05 15:32:16,459][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:32:16,460][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:32:17,274][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-320] due to args.save_total_limit |
| [2025-07-05 15:36:43,164][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:36:43,168][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:36:43,169][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:36:43,169][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:37:04,206][transformers.trainer][INFO] - Saving model checkpoint to /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-384 |
| [2025-07-05 15:37:04,529][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:37:04,530][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:37:05,339][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-352] due to args.save_total_limit |
| [2025-07-05 15:37:05,365][transformers.trainer][INFO] - |
|
|
| Training completed. Do not forget to share your model on huggingface.co/models =) |
|
|
|
|
| [2025-07-05 15:37:05,365][transformers.trainer][INFO] - Loading best model from /workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-224 (score: 0.5737133483698995). |
| [2025-07-05 15:37:05,576][transformers.trainer][INFO] - Deleting older checkpoint [/workspace/jbcs2025/outputs/2025-07-05/14-38-41/results/checkpoint-384] due to args.save_total_limit |
| [2025-07-05 15:37:05,608][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:37:05,612][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:37:05,612][transformers.trainer][INFO] - Num examples = 132 |
| [2025-07-05 15:37:05,612][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:37:26,645][__main__][INFO] - Training completed successfully. |
| [2025-07-05 15:37:26,645][__main__][INFO] - Running on Test |
| [2025-07-05 15:37:26,645][transformers.trainer][INFO] - The following columns in the Evaluation set don't have a corresponding argument in `PeftModelForSequenceClassification.forward` and have been ignored: prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt. If prompt, id, grades, reference, supporting_text, essay_text, essay_year, id_prompt are not expected by `PeftModelForSequenceClassification.forward`, you can safely ignore this message. |
| [2025-07-05 15:37:26,648][transformers.trainer][INFO] - |
| ***** Running Evaluation ***** |
| [2025-07-05 15:37:26,648][transformers.trainer][INFO] - Num examples = 138 |
| [2025-07-05 15:37:26,648][transformers.trainer][INFO] - Batch size = 4 |
| [2025-07-05 15:37:48,823][__main__][INFO] - Test metrics: {'eval_loss': 1.0317658185958862, 'eval_model_preparation_time': 0.0117, 'eval_accuracy': 0.6956521739130435, 'eval_RMSE': 24.07717061715384, 'eval_QWK': 0.658956109134045, 'eval_HDIV': 0.007246376811594235, 'eval_Macro_F1': 0.4583747484207799, 'eval_Micro_F1': 0.6956521739130435, 'eval_Weighted_F1': 0.6809435437313924, 'eval_TP_0': 0, 'eval_TN_0': 137, 'eval_FP_0': 0, 'eval_FN_0': 1, 'eval_TP_1': 0, 'eval_TN_1': 138, 'eval_FP_1': 0, 'eval_FN_1': 0, 'eval_TP_2': 7, 'eval_TN_2': 125, 'eval_FP_2': 3, 'eval_FN_2': 3, 'eval_TP_3': 52, 'eval_TN_3': 56, 'eval_FP_3': 16, 'eval_FN_3': 14, 'eval_TP_4': 36, 'eval_TN_4': 67, 'eval_FP_4': 20, 'eval_FN_4': 15, 'eval_TP_5': 1, 'eval_TN_5': 125, 'eval_FP_5': 3, 'eval_FN_5': 9, 'eval_runtime': 22.1629, 'eval_samples_per_second': 6.227, 'eval_steps_per_second': 1.579, 'epoch': 12.0} |
| [2025-07-05 15:37:48,824][transformers.trainer][INFO] - Saving model checkpoint to ./results/best_model |
| [2025-07-05 15:37:49,128][transformers.configuration_utils][INFO] - loading configuration file config.json from cache at /workspace/.hf_home/hub/models--meta-llama--Llama-3.1-8B/snapshots/d04e592bb4f6aa9cfee91e2e20afa771667e1d4b/config.json |
| [2025-07-05 15:37:49,129][transformers.configuration_utils][INFO] - Model config LlamaConfig { |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 128000, |
| "eos_token_id": 128001, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_position_embeddings": 131072, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.53.0", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
|
|
| [2025-07-05 15:37:49,499][transformers.tokenization_utils_base][INFO] - tokenizer config file saved in ./results/best_model/tokenizer_config.json |
| [2025-07-05 15:37:49,499][transformers.tokenization_utils_base][INFO] - Special tokens file saved in ./results/best_model/special_tokens_map.json |
| [2025-07-05 15:37:49,623][__main__][INFO] - Model and tokenizer saved to ./results/best_model |
| [2025-07-05 15:37:49,648][__main__][INFO] - Fine Tuning Finished. |
| [2025-07-05 15:37:50,159][__main__][INFO] - Total emissions: 0.0451 kg CO2eq |
|
|
