INNER CODE UNIT · Python

set_batch_size

tensorlayer/HyperPose · hyperpose/Config/__init__.py:342

def set_batch_size(batch_size):
    '''set the batch size in training

    Parameters
    ----------
    arg1 : int
        batch_size
    
    Returns
    -------
    None
    '''
    update_train.batch_size=batch_size


def set_kungfu_option(kungfu_option):
    '''set the optimizor of parallel training

View source record →