Training in progress, step 50

adapter_config.json

@@ -0,0 +1,31 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "alexsherstinsky/Mistral-7B-v0.1-sharded",
+  "bias": "none",
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 32,
+  "lora_dropout": 0.1,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 8,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "q_proj",
+    "gate_proj",
+    "o_proj",
+    "v_proj",
+    "up_proj",
+    "down_proj",
+    "k_proj"
+  ],
+  "task_type": "CAUSAL_LM"
+}

adapter_model.safetensors

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

runs/Jan22_03-50-11_8b33d498b7ab/events.out.tfevents.1705895557.8b33d498b7ab.1716.0

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

training_args.bin

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

training_config.json

@@ -0,0 +1,93 @@
+{
+  "experiment_key": "base",
+  "save_safetensors": true,
+  "max_shard_size": "10GB",
+  "local_rank": 0,
+  "use_gradient_checkpointing": true,
+  "trainer_key": "lm",
+  "force_fp32": false,
+  "force_fp16": false,
+  "from_gptq": false,
+  "huggingface_hub_token": null,
+  "single_gpu": true,
+  "master_port": 9994,
+  "deepspeed_stage": "stage_2",
+  "deepspeed_config_path": null,
+  "fsdp_strategy": "",
+  "fsdp_offload": true,
+  "seed": 42,
+  "stabilize": false,
+  "norm_fp32": false,
+  "path_to_env_file": "./.env",
+  "prepare_dataset": true,
+  "lora_hub_model_id": null,
+  "lora_model_local_path": null,
+  "fused_model_local_path": null,
+  "fuse_after_training": false,
+  "quantization_dataset_id": null,
+  "quantization_max_samples": 1024,
+  "quantized_model_path": "./quantized_model/",
+  "quantized_hub_model_id": null,
+  "quantized_hub_private_repo": true,
+  "dataset_key": "soda",
+  "train_local_path_to_data": "./train.jsonl",
+  "eval_local_path_to_data": null,
+  "shuffle": true,
+  "max_eval_samples": 1000,
+  "add_eval_to_train_if_no_path": false,
+  "tokenizer_name_or_path": null,
+  "tokenizer_use_fast": null,
+  "tokenizer_padding_side": null,
+  "collator_key": "lm",
+  "max_length": 2048,
+  "model_name_or_path": "alexsherstinsky/Mistral-7B-v0.1-sharded",
+  "push_to_hub_bos_add_bos_token": false,
+  "use_flash_attention_2": false,
+  "trust_remote_code": true,
+  "device_map": null,
+  "prepare_model_for_kbit_training": true,
+  "offload_folder": null,
+  "load_in_8bit": false,
+  "load_in_4bit": true,
+  "llm_int8_threshold": 6.0,
+  "llm_int8_has_fp16_weight": true,
+  "bnb_4bit_use_double_quant": true,
+  "bnb_4bit_quant_type": "nf4",
+  "bnb_quantize_after_model_init": false,
+  "gptq_bits": 4,
+  "gptq_group_size": 128,
+  "gptq_disable_exllama": true,
+  "apply_lora": true,
+  "lora_rank": 8,
+  "lora_alpha": 32,
+  "lora_dropout": 0.1,
+  "raw_lora_target_modules": "all",
+  "output_dir": "./outputs/",
+  "per_device_train_batch_size": 2,
+  "do_eval": false,
+  "per_device_eval_batch_size": null,
+  "gradient_accumulation_steps": 2,
+  "eval_accumulation_steps": null,
+  "eval_delay": 0,
+  "eval_steps": 1000,
+  "warmup_steps": 5,
+  "max_steps": 101,
+  "num_train_epochs": 1,
+  "learning_rate": 0.0002,
+  "max_grad_norm": 1.0,
+  "weight_decay": 0.001,
+  "label_smoothing_factor": 0.0,
+  "logging_steps": 1,
+  "save_steps": 50,
+  "save_total_limit": 1,
+  "optim": "paged_adamw_8bit",
+  "push_to_hub": true,
+  "hub_model_id": "mogaio/Mistral-2x2_8-LoRA",
+  "hub_private_repo": false,
+  "neftune_noise_alpha": null,
+  "project_name": null,
+  "report_to_wandb": false,
+  "wandb_api_key": null,
+  "wandb_project": "Mistral-hh-rlhf-fine-tuning",
+  "wandb_entity": "Mistral-hh-rlhf"
+}