示例#1
0
class TextEncoderConfig(metaclass=ABCMeta):
    pad_index = abstract_attribute()
    embed_size = abstract_attribute()
    hidden_size = abstract_attribute()
    num_directions = abstract_attribute()
    num_layers = abstract_attribute()
    dropout = abstract_attribute()
    text_feat_size = abstract_attribute()
    vocab_size = abstract_attribute()
    embed_init = abstract_attribute()
示例#2
0
class TrainConfig(metaclass=ABCMeta):
    batch_size = abstract_attribute()
    num_iterations = abstract_attribute()
    learning_rate = abstract_attribute()
    print_freq = abstract_attribute()
    valid_freq = abstract_attribute()
    patience = abstract_attribute()
    num_data_loader_workers = abstract_attribute()
示例#3
0
class ImageEncoderConfig(metaclass=ABCMeta):
    image_size = abstract_attribute()
    num_channels = abstract_attribute()
    text_feat_size = abstract_attribute()
    image_feat_size = abstract_attribute()
class TextDecoderConfig(metaclass=ABCMeta):
    text_length = abstract_attribute()
    pad_index = abstract_attribute()
    word_embed_size = abstract_attribute()
    embed_size = abstract_attribute()
    hidden_size = abstract_attribute()
    num_directions = abstract_attribute()
    num_layers = abstract_attribute()
    dropout = abstract_attribute()
    to_hidden_fc_sizes = abstract_attribute()
    vocab_size = abstract_attribute()
    embed_init = abstract_attribute()
    utter_type_size = abstract_attribute()
    intention_embed_size = abstract_attribute()
示例#5
0
class TestConfig(metaclass=ABCMeta):
    batch_size = abstract_attribute()
    num_data_loader_workers = abstract_attribute()
示例#6
0
class MemoryConfig(metaclass=ABCMeta):
    knowledge_size = abstract_attribute()
    entry_size = abstract_attribute()
    memory_size = 512
    output_size = 512