| { |
| "save_dir_name": "NoPromptV2_BEST", |
| "load_trained_model_path": "saved_models/VisualGPT_NoPrompt_v2_best.pth", |
| "dataset_name": "MSR-VTT", |
| "model_name": "gpt2", |
| "exp_name": "VisualGPT_NoPrompt", |
| "log_file": "log/test/VisualGPT_NoPrompt_v2_epoch7.txt", |
| "data_dir_path": "data", |
| "feature_type": "CLIP", |
| "clip_pretrained_path": "ViT-B/32", |
| "num_target_captions": 10, |
| "max_sequence_length": 25, |
| "max_frames": 25, |
| "tfm_heads": 8, |
| "tfm_layers": 3, |
| "batch_size": 32, |
| "num_workers": 4, |
| "lr": 0.0001, |
| "random_seed": 42, |
| "max_grad_norm": 1.0, |
| "huggingface_model_name": "trained_model.pth" |
| } |