config.py

from jobman import DD

RAB_DATASET_BASE_PATH = '/mnt/disk2/guozhao/predatas/youtube/'
RAB_FEATURE_BASE_PATH = '/mnt/disk2/guozhao/features/youtube/inception-v3'
RAB_EXP_PATH = '/home/guozhao/results/zhao/'

config = DD({
    'model': 'attention',
    'random_seed': 1234,
    # ERASE everything under save_model_path
    'erase_history': True,
    'attention': DD({
        'reload_': False,
        'verbose': True,
        'debug': False,
        'save_model_dir': RAB_EXP_PATH + 'save_dir/',
        'from_dir': RAB_EXP_PATH + 'from_dir/',
        # dataset
        'dataset': 'youtube2text',
        'video_feature': 'googlenet',
        'K':28, # 26 when compare
        'OutOf':None,
        # network
        'dim_word':512,#468, # 474
        'ctx_dim':-1,# auto set
        'dim':512, #1024, # lstm dim # 536
        'n_layers_out':1, # for predicting next word
        'n_layers_init':0,
        'encoder_dim': 512,#300,
        'prev2out':True,
        'ctx2out':True,
        'selector':True,
        'n_words':20000,
        'maxlen':30, # max length of the descprition
        'use_dropout':True,
        'isGlobal': False,
        # training
        'patience':20,
        'max_epochs':500,
        'decay_c':1e-4,
        'alpha_entropy_r': 0.,
        'alpha_c':0.70602,
        'lrate':0.0001,
        'optimizer':'adadelta',
        'clip_c': 10.,
        # minibatches
        'batch_size': 64, # for trees use 25
        'valid_batch_size':200,
        'dispFreq':10,
        'validFreq':2000,
        'saveFreq':-1, # this is disabled, now use sampleFreq instead
        'sampleFreq':100,
        # blue, meteor, or both
        'metric': 'everything', # set to perplexity on DVS
        }),
    })