Rename 24_10_20_08_23_42_pretrain17_finetune3_log.txt to 24_10_20_08_23_42_pretrain18_finetune3_log.txt
10ccb74
verified
{ | |
"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"} | |