Python resnet_model.resnet_v1() Examples

The following are 1 code examples of resnet_model.resnet_v1(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module resnet_model , or try the search function .
Example #1
Source File: resnet_main.py    From DistributedDeepLearning with MIT License 6 votes vote down vote up 
def build_network(features, mode, params):
    """ Build ResNet50 Model

    Args:
        features:
        mode:
        params:

    Returns:
        Model function
    """
    network = resnet_v1(
        resnet_depth=50,
        num_classes=params["classes"],
        data_format=params["data_format"],
    )
    return network(inputs=features, is_training=(mode == tf.estimator.ModeKeys.TRAIN))