{
    "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": 5,
        "iters_per_epoch": 200,
        "num_workers": 4,
        "warmup_steps": 200,
        "seed": 42,
        "output_dir": "output/minigpt4_stage2_finetune",
        "amp": true,
        "resume_ckpt_path": null,
        "evaluate": false,
        "train_splits": [
            "train"
        ],
        "device": "cuda",
        "world_size": 1,
        "dist_url": "env://",
        "distributed": true,
        "wandb_log": true,
        "job_name": "minigpt4_finetune",
        "rank": 0,
        "gpu": 0,
        "dist_backend": "nccl"
    },
    "model": {
        "arch": "minigpt4",
        "image_size": 224,
        "drop_path_rate": 0,
        "use_grad_checkpoint": false,
        "vit_precision": "fp16",
        "freeze_vit": true,
        "freeze_qformer": true,
        "num_query_token": 32,
        "prompt": "",
        "llama_model": "/home1/gmh/Cheetah/vicuna-7b",
        "model_type": "pretrain_vicuna0",
        "max_txt_len": 160,
        "end_sym": "###",
        "prompt_path": "prompts/alignment.txt",
        "prompt_template": "###Human: {} ###Assistant: ",
        "ckpt": "/home1/gmh/MiniGPT4/MiniGPT-4/prerained_minigpt4_7b.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": {
        "cc_sbu_align": {
            "data_type": "images",
            "build_info": {
                "storage": "/home1/gmh/MiniGPT4/cc_sbu_align"
            },
            "batch_size": 12,
            "train_dataset_ratio": 0.5,
            "vis_processor": {
                "train": {
                    "name": "blip2_image_train",
                    "image_size": 224
                }
            },
            "text_processor": {
                "train": {
                    "name": "blip_caption"
                }
            }
        },
        "cc_sbu_cot": {
            "data_type": "images",
            "build_info": {
                "storage": "/home1/gmh/MiniGPT4/cc_sbu_cot"
            },
            "batch_size": 12,
            "train_dataset_ratio": 0.5,
            "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": 5,
        "iters_per_epoch": 200,
        "num_workers": 4,
        "warmup_steps": 200,
        "seed": 42,
        "output_dir": "output/minigpt4_stage2_finetune",
        "amp": true,
        "resume_ckpt_path": null,
        "evaluate": false,
        "train_splits": [
            "train"
        ],
        "device": "cuda",
        "world_size": 1,
        "dist_url": "env://",
        "distributed": true,
        "wandb_log": true,
        "job_name": "minigpt4_finetune",
        "rank": 0,
        "gpu": 0,
        "dist_backend": "nccl"
    },
    "model": {
        "arch": "minigpt4",
        "image_size": 224,
        "drop_path_rate": 0,
        "use_grad_checkpoint": false,
        "vit_precision": "fp16",
        "freeze_vit": true,
        "freeze_qformer": true,
        "num_query_token": 32,
        "prompt": "",
        "llama_model": "/home1/gmh/Cheetah/vicuna-7b",
        "model_type": "pretrain_vicuna0",
        "max_txt_len": 160,
        "end_sym": "###",
        "prompt_path": "prompts/alignment.txt",
        "prompt_template": "###Human: {} ###Assistant: ",
        "ckpt": "/home1/gmh/MiniGPT4/MiniGPT-4/prerained_minigpt4_7b.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": {
        "cc_sbu_align": {
            "data_type": "images",
            "build_info": {
                "storage": "/home1/gmh/MiniGPT4/cc_sbu_align"
            },
            "batch_size": 12,
            "train_dataset_ratio": 0.5,
            "vis_processor": {
                "train": {
                    "name": "blip2_image_train",
                    "image_size": 224
                }
            },
            "text_processor": {
                "train": {
                    "name": "blip_caption"
                }
            }
        },
        "cc_sbu_cot": {
            "data_type": "images",
            "build_info": {
                "storage": "/home1/gmh/MiniGPT4/cc_sbu_cot"
            },
            "batch_size": 12,
            "train_dataset_ratio": 0.5,
            "vis_processor": {
                "train": {
                    "name": "blip2_image_train",
                    "image_size": 224
                }
            },
            "text_processor": {
                "train": {
                    "name": "blip_caption"
                }
            }
        }
    }
}
