def valid_input_fn()

in src/train.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

def valid_input_fn(hvd, mpi=False):
    if mpi:
        return _input(args.epochs, args.batch_size, None, 'valid', hvd)
    else:
        return _input(args.epochs, args.batch_size, args.valid, 'valid')