Revise the use_xla_jit option.
Revise it such that the option specified in the config is also respected,
although the argument can override it.
@@ -524,7 +524,8 @@ def get_base_config(args):
if base_config is None:
raise ValueError('base_config dictionary has to be '
'defined in the config file')
- base_config['use_xla_jit'] = args.use_xla_jit
+ base_config['use_xla_jit'] = args.use_xla_jit or base_config.get('use_xla_jit', False)
+
base_model = config_module.get('base_model', None)
if base_model is None:
