{ "run": { "task": "image_text_pretrain", "lr_sched": "linear_warmup_cosine_lr", "init_lr": 3e-05, "min_lr": 1e-05, "warmup_lr": 1e-06, "weight_decay": 0.05, "max_epoch": 1, "iters_per_epoch": 2501, "batch_size_train": 20, "batch_size_eval": 1, "num_workers": 4, "warmup_steps": 200, "seed": 42, "output_dir": "output/xraygpt_openi_finetune", "amp": true, "resume_ckpt_path": null, "evaluate": false, "train_splits": [ "train" ], "device": "cuda", "world_size": 1, "dist_url": "env://", "distributed": true, "rank": 0, "gpu": 0, "dist_backend": "nccl" }, "model": { "arch": "mini_gpt4", "image_size": 224, "drop_path_rate": 0, "use_grad_checkpoint": false, "vit_precision": "fp16", "freeze_vit": true, "freeze_qformer": true, "num_query_token": 32, "llama_model": "Joycean0301/Llama-3.2-3B-Instruct-Medical-Conversational", "prompt": "", "model_type": "pretrain_vicuna", "max_txt_len": 160, "end_sym": "<|eot_id|>", "prompt_path": "prompts/alignment.txt", "prompt_template": "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nIf you are a doctor, please answer the medical questions based on chest xray image and the patient's description.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n ", "ckpt": "24_10_19_23_35_50_epoch17.pth" }, "preprocess": { "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 }, "eval": { "name": "blip2_image_eval", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" }, "eval": { "name": "blip_caption" } } }, "datasets": { "openi": { "data_type": "images", "build_info": { "storage": "dataset/openi" }, "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" } } } } } { "run": { "task": "image_text_pretrain", "lr_sched": "linear_warmup_cosine_lr", "init_lr": 3e-05, "min_lr": 1e-05, "warmup_lr": 1e-06, "weight_decay": 0.05, "max_epoch": 1, "iters_per_epoch": 2501, "batch_size_train": 20, "batch_size_eval": 1, "num_workers": 4, "warmup_steps": 200, "seed": 42, "output_dir": "output/xraygpt_openi_finetune", "amp": true, "resume_ckpt_path": null, "evaluate": false, "train_splits": [ "train" ], "device": "cuda", "world_size": 1, "dist_url": "env://", "distributed": true, "rank": 0, "gpu": 0, "dist_backend": "nccl" }, "model": { "arch": "mini_gpt4", "image_size": 224, "drop_path_rate": 0, "use_grad_checkpoint": false, "vit_precision": "fp16", "freeze_vit": true, "freeze_qformer": true, "num_query_token": 32, "llama_model": "Joycean0301/Llama-3.2-3B-Instruct-Medical-Conversational", "prompt": "", "model_type": "pretrain_vicuna", "max_txt_len": 160, "end_sym": "<|eot_id|>", "prompt_path": "prompts/alignment.txt", "prompt_template": "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nIf you are a doctor, please answer the medical questions based on chest xray image and the patient's description.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n ", "ckpt": "24_10_19_23_35_50_epoch17.pth" }, "preprocess": { "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 }, "eval": { "name": "blip2_image_eval", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" }, "eval": { "name": "blip_caption" } } }, "datasets": { "openi": { "data_type": "images", "build_info": { "storage": "dataset/openi" }, "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" } } } } } { "run": { "task": "image_text_pretrain", "lr_sched": "linear_warmup_cosine_lr", "init_lr": 3e-05, "min_lr": 1e-05, "warmup_lr": 1e-06, "weight_decay": 0.05, "max_epoch": 1, "iters_per_epoch": 2501, "batch_size_train": 25, "batch_size_eval": 1, "num_workers": 4, "warmup_steps": 200, "seed": 42, "output_dir": "output/xraygpt_openi_finetune", "amp": true, "resume_ckpt_path": null, "evaluate": false, "train_splits": [ "train" ], "device": "cuda", "world_size": 1, "dist_url": "env://", "distributed": true, "rank": 0, "gpu": 0, "dist_backend": "nccl" }, "model": { "arch": "mini_gpt4", "image_size": 224, "drop_path_rate": 0, "use_grad_checkpoint": false, "vit_precision": "fp16", "freeze_vit": true, "freeze_qformer": true, "num_query_token": 32, "llama_model": "Joycean0301/Llama-3.2-3B-Instruct-Medical-Conversational", "prompt": "", "model_type": "pretrain_vicuna", "max_txt_len": 160, "end_sym": "<|eot_id|>", "prompt_path": "prompts/alignment.txt", "prompt_template": "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nIf you are a doctor, please answer the medical questions based on chest xray image and the patient's description.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n ", "ckpt": "24_10_19_23_35_50_epoch17.pth" }, "preprocess": { "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 }, "eval": { "name": "blip2_image_eval", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" }, "eval": { "name": "blip_caption" } } }, "datasets": { "openi": { "data_type": "images", "build_info": { "storage": "dataset/openi" }, "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" } } } } } { "run": { "task": "image_text_pretrain", "lr_sched": "linear_warmup_cosine_lr", "init_lr": 3e-05, "min_lr": 1e-05, "warmup_lr": 1e-06, "weight_decay": 0.05, "max_epoch": 1, "iters_per_epoch": 2501, "batch_size_train": 30, "batch_size_eval": 1, "num_workers": 4, "warmup_steps": 200, "seed": 42, "output_dir": "output/xraygpt_openi_finetune", "amp": true, "resume_ckpt_path": null, "evaluate": false, "train_splits": [ "train" ], "device": "cuda", "world_size": 1, "dist_url": "env://", "distributed": true, "rank": 0, "gpu": 0, "dist_backend": "nccl" }, "model": { "arch": "mini_gpt4", "image_size": 224, "drop_path_rate": 0, "use_grad_checkpoint": false, "vit_precision": "fp16", "freeze_vit": true, "freeze_qformer": true, "num_query_token": 32, "llama_model": "Joycean0301/Llama-3.2-3B-Instruct-Medical-Conversational", "prompt": "", "model_type": "pretrain_vicuna", "max_txt_len": 160, "end_sym": "<|eot_id|>", "prompt_path": "prompts/alignment.txt", "prompt_template": "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nIf you are a doctor, please answer the medical questions based on chest xray image and the patient's description.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n ", "ckpt": "24_10_19_23_35_50_epoch17.pth" }, "preprocess": { "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 }, "eval": { "name": "blip2_image_eval", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" }, "eval": { "name": "blip_caption" } } }, "datasets": { "openi": { "data_type": "images", "build_info": { "storage": "dataset/openi" }, "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" } } } } } { "run": { "task": "image_text_pretrain", "lr_sched": "linear_warmup_cosine_lr", "init_lr": 3e-05, "min_lr": 1e-05, "warmup_lr": 1e-06, "weight_decay": 0.05, "max_epoch": 1, "iters_per_epoch": 2501, "batch_size_train": 25, "batch_size_eval": 1, "num_workers": 4, "warmup_steps": 200, "seed": 42, "output_dir": "output/xraygpt_openi_finetune", "amp": true, "resume_ckpt_path": null, "evaluate": false, "train_splits": [ "train" ], "device": "cuda", "world_size": 1, "dist_url": "env://", "distributed": true, "rank": 0, "gpu": 0, "dist_backend": "nccl" }, "model": { "arch": "mini_gpt4", "image_size": 224, "drop_path_rate": 0, "use_grad_checkpoint": false, "vit_precision": "fp16", "freeze_vit": true, "freeze_qformer": true, "num_query_token": 32, "llama_model": "Joycean0301/Llama-3.2-3B-Instruct-Medical-Conversational", "prompt": "", "model_type": "pretrain_vicuna", "max_txt_len": 160, "end_sym": "<|eot_id|>", "prompt_path": "prompts/alignment.txt", "prompt_template": "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nIf you are a doctor, please answer the medical questions based on chest xray image and the patient's description.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n ", "ckpt": "24_10_19_23_35_50_epoch17.pth" }, "preprocess": { "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 }, "eval": { "name": "blip2_image_eval", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" }, "eval": { "name": "blip_caption" } } }, "datasets": { "openi": { "data_type": "images", "build_info": { "storage": "dataset/openi" }, "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" } } } } } { "run": { "task": "image_text_pretrain", "lr_sched": "linear_warmup_cosine_lr", "init_lr": 3e-05, "min_lr": 1e-05, "warmup_lr": 1e-06, "weight_decay": 0.05, "max_epoch": 3, "iters_per_epoch": 3000, "batch_size_train": 25, "batch_size_eval": 1, "num_workers": 4, "warmup_steps": 200, "seed": 42, "output_dir": "output/xraygpt_openi_finetune", "amp": true, "resume_ckpt_path": null, "evaluate": false, "train_splits": [ "train" ], "device": "cuda", "world_size": 1, "dist_url": "env://", "distributed": true, "rank": 0, "gpu": 0, "dist_backend": "nccl" }, "model": { "arch": "mini_gpt4", "image_size": 224, "drop_path_rate": 0, "use_grad_checkpoint": false, "vit_precision": "fp16", "freeze_vit": true, "freeze_qformer": true, "num_query_token": 32, "llama_model": "Joycean0301/Llama-3.2-3B-Instruct-Medical-Conversational", "prompt": "", "model_type": "pretrain_vicuna", "max_txt_len": 160, "end_sym": "<|eot_id|>", "prompt_path": "prompts/alignment.txt", "prompt_template": "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nIf you are a doctor, please answer the medical questions based on chest xray image and the patient's description.<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n ", "ckpt": "24_10_19_23_35_50_epoch17.pth" }, "preprocess": { "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 }, "eval": { "name": "blip2_image_eval", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" }, "eval": { "name": "blip_caption" } } }, "datasets": { "openi": { "data_type": "images", "build_info": { "storage": "dataset/openi" }, "vis_processor": { "train": { "name": "blip2_image_train", "image_size": 224 } }, "text_processor": { "train": { "name": "blip_caption" } } } } } {"train_lr": "0.000", "train_loss": "1.346"} {"train_lr": "0.000", "train_loss": "1.249"} {"train_lr": "0.000", "train_loss": "1.220"}