コード例 #1
0
ファイル: text_encoder_config.py プロジェクト: chorseng/MAGIC
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()
コード例 #4
0
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
ファイル: memory_config.py プロジェクト: chorseng/MAGIC
class MemoryConfig(metaclass=ABCMeta):
    knowledge_size = abstract_attribute()
    entry_size = abstract_attribute()
    memory_size = 512
    output_size = 512