Fix error in onmt-main when using run types other than train (#648)

OpenNMT · Apr 14, 2020 · 45d3f82 · 45d3f82
1 parent f14c05a
commit 45d3f82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opennmt/bin/main.py b/opennmt/bin/main.py
@@ -176,7 +176,7 @@ def main():
   tf.config.threading.set_inter_op_parallelism_threads(args.inter_op_parallelism_threads)
 
   gpus = tf.config.list_physical_devices(device_type="GPU")
-  if args.horovod:
+  if hasattr(args, "horovod") and args.horovod:
     import horovod.tensorflow as hvd  # pylint: disable=import-outside-toplevel
     hvd.init()
     is_master = hvd.rank() == 0