Python ResnetModelConfig примеры использования

Пример #1

Файл: cifar_train.py Проект: zylprivate/learning-to-reweight-examples

def _get_config():
    """Gets config."""
    config_file = FLAGS.config
    config = ResnetModelConfig()
    assert config_file is not None, 'Must pass in a configuration file through --config'
    Merge(open(config_file).read(), config)
    return config

Пример #2

def _get_config():
    """Gets config."""
    if FLAGS.config is None:
        config_file = os.path.join(
            os.path.realpath(cnn.__path__[0]), 'configs/cifar-{}.prototxt'.format(FLAGS.model))
    else:
        config_file = FLAGS.config
    config = ResnetModelConfig()
    Merge(open(config_file).read(), config)
    return config

Пример #3

Файл: resnet_model.py Проект: zylprivate/learning-to-reweight-examples

 def create_from_file(cls,
                      config_filename,
                      is_training=True,
                      inp=None,
                      label=None,
                      batch_size=None):
     config = ResnetModelConfig()
     Merge(open(config_filename).read(), config)
     return cls(config,
                is_training=is_training,
                inp=inp,
                label=label,
                batch_size=batch_size)

Пример #4

def _get_config():
    """Gets config."""
    config_file = 'cifar/configs/resnet-test.prototxt'
    config = ResnetModelConfig()
    Merge(open(config_file).read(), config)
    return config