
"""
class SIMMCFashionConfig():
    #? not used
    _FASHION_ACTION_NO = 5
    _FASHION_ATTRS_NO = 33
"""

model_conf = {
    'dropout_prob': 0.5,
    'freeze_bert': True,
    'n_decoders': 2,
    'decoder_heads': 6 #todo must be a perfect divisor of 768
}

special_toks = {
    'pad_token': '[PAD]',
    'start_token': '[CLS]',
    'end_token': '[SEP]',
    'unk_token': '[UNK]',
}

train_conf = {
    'seed': 240797,
    'distractors_sampling': -1, #-1 to avoid sampling
    'lr': 1e-3,
    'weight_decay': 1e-3,
    'ckpt_folder': 'mm_response_generation/checkpoints'
}

"""
class SIMMCFashionConfig():
    #? not used
    _FASHION_ACTION_NO = 5
    _FASHION_ATTRS_NO = 33
"""

model_conf = {
    'dropout_prob': 0.5,
    'freeze_bert': True,
    'n_decoders': 2,
    'decoder_heads': 6 #todo must be a perfect divisor of 768
}

special_toks = {
    'pad_token': '[PAD]',
    'start_token': '[CLS]',
    'end_token': '[SEP]',
    'unk_token': '[UNK]',
}

train_conf = {
    'seed': 240797,
    'distractors_sampling': -1, #-1 to avoid sampling
    'lr': 1e-3,
    'weight_decay': 1e-3,
    'ckpt_folder': 'mm_response_generation/checkpoints'
}
class SIMMCFashionConfig():     #? not used     _FASHION_ACTION_NO = 5     _FASHION_ATTRS_NO = 33 #todo must be a perfect divisor of 768 #-1 to avoid sampling