coco-gpt_medium-en_clip_b32-transformer-finetune-ep10-bs48-lr2e-05/args.json

{
    "train_name_prefix": null,
    "dataset_name": "coco",
    "rinna_gpt_name": "gpt_medium",
    "clip_model_name": "en_clip_b32",
    "pretrained_path": null,
    "datasets_dpath": "./data",
    "checkpoints_dpath": "./checkpoints",
    "epochs": 10,
    "per_gpu_train_batch_size": 24,
    "per_gpu_eval_batch_size": 24,
    "lr": 2e-05,
    "warmup_steps": 5000,
    "save_every": 1,
    "mapping_type": "transformer",
    "prefix_length": 10,
    "prefix_length_clip": 10,
    "only_prefix": false,
    "num_layers": 8,
    "n_gpu": 2,
    "prefix_dim": 512,
    "train_batch_size": 48,
    "eval_batch_size": 48,
    "train_data_fpath": "./data/coco/processed-en_clip_b32/train.pkl",
    "valid_data_fpath": "./data/coco/processed-en_clip_b32/valid.pkl"
}