Pretrain_LLaMA3.2/log_epoch100_160.txt

{
    "run": {
        "task": "image_text_pretrain",
        "lr_sched": "linear_warmup_cosine_lr",
        "init_lr": 0.0001,
        "min_lr": 8e-05,
        "warmup_lr": 1e-06,
        "weight_decay": 0.05,
        "max_epoch": 150,
        "batch_size_train": 120,
        "batch_size_eval": 3,
        "num_workers": 4,
        "warmup_steps": 500,
        "iters_per_epoch": 500,
        "seed": 42,
        "output_dir": "output/xraygpt_mimic_pretrain",
        "amp": true,
        "resume_ckpt_path": "24_10_22_08_14_39_epoch99.pth",
        "evaluate": false,
        "train_splits": [
            "train"
        ],
        "device": "cuda",
        "world_size": 4,
        "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"
    },
    "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": {
        "mimic": {
            "data_type": "images",
            "build_info": {
                "storage": "dataset/mimic"
            },
            "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": 0.0001,
        "min_lr": 8e-05,
        "warmup_lr": 1e-06,
        "weight_decay": 0.05,
        "max_epoch": 150,
        "batch_size_train": 120,
        "batch_size_eval": 3,
        "num_workers": 4,
        "warmup_steps": 500,
        "iters_per_epoch": 500,
        "seed": 42,
        "output_dir": "output/xraygpt_mimic_pretrain",
        "amp": true,
        "resume_ckpt_path": "24_10_22_08_14_39_epoch99.pth",
        "evaluate": false,
        "train_splits": [
            "train"
        ],
        "device": "cuda",
        "world_size": 4,
        "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"
    },
    "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": {
        "mimic": {
            "data_type": "images",
            "build_info": {
                "storage": "dataset/mimic"
            },
            "vis_processor": {
                "train": {
                    "name": "blip2_image_train",
                    "image_size": 224
                }
            },
            "text_processor": {
                "train": {
                    "name": "blip_caption"
                }
            }
        }
    }
}
{"train_lr": "0.000085", "train_loss": "1.176420"}
{"train_lr": "0.000085", "train_loss": "1.175911"}
{"train_lr": "0.000085", "train_loss": "1.177007"}
{"train_lr": "0.000084", "train_loss": "1.174899"}
{"train_lr": "0.000084", "train_loss": "1.176167"}
{"train_lr": "0.000084", "train_loss": "1.175371"}
{"train_lr": "0.000084", "train_loss": "1.175089"}
{"train_lr": "0.000084", "train_loss": "1.174685"}
{"train_lr": "0.000084", "train_loss": "1.173965"}
{"train_lr": "0.000083", "train_loss": "1.173050"}
{"train_lr": "0.000083", "train_loss": "1.174454"}
{"train_lr": "0.000083", "train_loss": "1.174522"}
{"train_lr": "0.000083", "train_loss": "1.171317"}
{"train_lr": "0.000083", "train_loss": "1.173012"}
{"train_lr": "0.000083", "train_loss": "1.173456"}
{"train_lr": "0.000082", "train_loss": "1.172261"}
{"train_lr": "0.000082", "train_loss": "1.171457"}
{"train_lr": "0.000082", "train_loss": "1.172554"}
{"train_lr": "0.000082", "train_loss": "1.169906"}
{"train_lr": "0.000082", "train_loss": "1.171817"}
{"train_lr": "0.000082", "train_loss": "1.170531"}
{"train_lr": "0.000082", "train_loss": "1.170232"}
{"train_lr": "0.000082", "train_loss": "1.171015"}
{"train_lr": "0.000082", "train_loss": "1.169610"}
{"train_lr": "0.000081", "train_loss": "1.170178"}
{"train_lr": "0.000081", "train_loss": "1.169042"}
{"train_lr": "0.000081", "train_loss": "1.170149"}
{"train_lr": "0.000081", "train_loss": "1.168049"}
{"train_lr": "0.000081", "train_loss": "1.169198"}
{"train_lr": "0.000081", "train_loss": "1.169054"}
{"train_lr": "0.000081", "train_loss": "1.168359"}
{"train_lr": "0.000081", "train_loss": "1.168657"}
{"train_lr": "0.000081", "train_loss": "1.168694"}
{"train_lr": "0.000081", "train_loss": "1.168019"}
{"train_lr": "0.000081", "train_loss": "1.167548"}
{"train_lr": "0.000080", "train_loss": "1.165927"}
{"train_lr": "0.000080", "train_loss": "1.168800"}v
{"train_lr": "0.000084", "train_loss": "1.166641"}
{"train_lr": "0.000084", "train_loss": "1.165976"}
{"train_lr": "0.000084", "train_loss": "1.167195"}
{"train_lr": "0.000084", "train_loss": "1.165221"}
{"train_lr": "0.000084", "train_loss": "1.166641"}
{"train_lr": "0.000084", "train_loss": "1.165935"}
{"train_lr": "0.000084", "train_loss": "1.165807"}
{"train_lr": "0.000084", "train_loss": "1.165503"}
{"train_lr": "0.000083", "train_loss": "1.164865"}
{"train_lr": "0.000083", "train_loss": "1.164062"}
{"train_lr": "0.000083", "train_loss": "1.165597"}
{"train_lr": "0.000083", "train_loss": "1.165696"}
{"train_lr": "0.000083", "train_loss": "1.162653"}
{"train_lr": "0.000083", "train_loss": "1.164436"}
{"train_lr": "0.000083", "train_loss": "1.164961"}
{"train_lr": "0.000083", "train_loss": "1.163872"}
{"train_lr": "0.000083", "train_loss": "1.163161"}
{"train_lr": "0.000082", "train_loss": "1.164320"}
{"train_lr": "0.000082", "train_loss": "1.161806"}
{"train_lr": "0.000082", "train_loss": "1.163766"}
{"train_lr": "0.000082", "train_loss": "1.162590"}
{"train_lr": "0.000082", "train_loss": "1.162393"}
{"train_lr": "0.000082", "train_loss": "1.163239"}
{"train_lr": "0.000082", "train_loss": "1.161929"}