Description : set gpu module

def setting_ctx(num_gpus):
    """
    Description : set gpu module
    """
    if num_gpus > 0:
        ctx = [mx.gpu(i) for i in range(num_gpus)]
    else:
        ctx = [mx.cpu()]
    return ctx