def _mp_fn(index, flags):
    global FLAGS
    FLAGS = flags
    torch.set_default_tensor_type('torch.FloatTensor')
    accuracy = train_imagenet()
    if (accuracy < FLAGS.target_accuracy):
        print('Accuracy {} is below target {}'.format(accuracy, FLAGS.target_accuracy))
        sys.exit(21)
def _mp_fn(index, flags):
    global FLAGS
    FLAGS = flags
    torch.set_default_tensor_type('torch.FloatTensor')
    accuracy = train_imagenet()
    if (accuracy < FLAGS.target_accuracy):
        print('Accuracy {} is below target {}'.format(accuracy, FLAGS.target_accuracy))
        sys.exit(21)