def get_training_context()

in mxnet_managed_spot_training_checkpointing/source_dir/mnist.py [0:0]

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

def get_training_context(num_gpus):
    if num_gpus:
        return [mx.gpu(i) for i in range(num_gpus)]
    else:
        return mx.cpu()