File size: 878 Bytes
8b742bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
{
"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"
} |