coco_based-sfcoco_merged-gpt_medium-en_clip_b32-transformer-finetune-ep20-bs8-lr2e-05/args.json

{
    "train_name_prefix": "coco_based",
    "dataset_name": "sfcoco_merged",
    "rinna_gpt_name": "gpt_medium",
    "clip_model_name": "en_clip_b32",
    "pretrained_path": "checkpoints/coco-gpt_medium-en_clip_b32-transformer-finetune-ep10-bs48-lr2e-05/004.pt",
    "datasets_dpath": "./data",
    "checkpoints_dpath": "./checkpoints",
    "epochs": 20,
    "per_gpu_train_batch_size": 8,
    "per_gpu_eval_batch_size": 4,
    "lr": 2e-05,
    "warmup_steps": 5000,
    "save_every": 2,
    "mapping_type": "transformer",
    "prefix_length": 10,
    "prefix_length_clip": 10,
    "only_prefix": false,
    "num_layers": 8,
    "n_gpu": 1,
    "prefix_dim": 512,
    "train_batch_size": 8,
    "eval_batch_size": 4,
    "train_data_fpath": "./data/sfcoco_merged/processed-en_clip_b32/train.pkl",
    "valid_data_fpath": "./data/sfcoco_merged/processed-en_clip_b32/valid.pkl"
}