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