Commit 94e0a473 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix argument check

parent f84eb0a1
...@@ -567,7 +567,7 @@ if __name__ == "__main__": ...@@ -567,7 +567,7 @@ if __name__ == "__main__":
(args.num_nodes is not None and args.num_nodes > 1))): (args.num_nodes is not None and args.num_nodes > 1))):
raise ValueError("For distributed training, --seed must be specified") raise ValueError("For distributed training, --seed must be specified")
if(args.precision == "16" and args.deepspeed_config_path is not None): if(str(args.precision) == "16" and args.deepspeed_config_path is not None):
raise ValueError("DeepSpeed and FP16 training are not compatible") raise ValueError("DeepSpeed and FP16 training are not compatible")
# This re-applies the training-time filters at the beginning of every epoch # This re-applies the training-time filters at the beginning of every epoch
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment