help='Whether to run in local mode (required)',
dest='local',
required=True)
parser.add_argument('-g',
'--gpu',
nargs=1,
type=str,
metavar='G',
help='Whether to run a GPU or CPU benchmark' +
' (required)',
dest='gpu',
required=True)
args = parser.parse_args()
print('Connecting to Cloudburst...')
cloudburst = CloudburstConnection(args.cloudburst[0], args.ip[0])
gpu = args.gpu[0].lower() == 'true'
if gpu:
german_init = english_to_german_init_gpu
french_init = english_to_french_init_gpu
german = english_to_german_gpu
french = english_to_french_gpu
else:
german_init = english_to_german_init
french_init = english_to_french_init
german = english_to_german
french = english_to_french
with open('imagenet_classes.txt', 'r') as f:
classes = [line.strip() for line in f.readlines()]