from jobman import DD
import common

exp_path = common.get_rab_exp_path()

config = DD({
    'model':
    'attention',
    'random_seed':
    1234,
    # ERASE everything under save_model_path
    'erase_history':
    True,
    'attention':
    DD({
        'reload_': False,
        'save_model_dir': exp_path + 'res_obj10+beam6/',
        'from_dir': '',
        'dataset': 'youtube2text',
        'video_feature': 'resnet152',
        'dim_word': 300,  # 474
        'ctx_dim': -1,  # auto set 
        'dim': 512,  # lstm dim # 536
        'n_layers_out': 1,  # for predicting next word        
        'n_layers_init': 0,
        'encoder_dim': 300,
        'prev2out': True,
        'ctx2out': True,
        'patience': 20,
        'max_epochs': 500,
        'decay_c': 1e-4,
Ejemplo n.º 2
0
from jobman import DD
import common

exp_path = common.get_rab_exp_path()

config = DD({
    'model': 'attention',
    'random_seed': 1234,
    # ERASE everything under save_model_path
    'erase_history': True,
    'attention': DD({
        'reload_': False,
        'save_model_dir': exp_path + 'arctic-capgen-vid/test_non/',
        'from_dir': '',
        'dataset': 'youtube2text',
        'video_feature': 'googlenet',
        'dim_word':468, # 474
        'ctx_dim':-1,# auto set 
        'dim':3518, # lstm dim # 536
        'n_layers_out':1, # for predicting next word        
        'n_layers_init':0, 
        'encoder_dim': 300,
        'prev2out':True, 
        'ctx2out':True, 
        'patience':20,
        'max_epochs':500, 
        'decay_c':1e-4,
        'alpha_entropy_r': 0.,
        'alpha_c':0.70602,
        'lrate':0.0001,
        'selector':True,