| { | |
| "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" | |
| } |