Update train.py

This commit is contained in:
LaviAvigad 2022-04-05 13:53:16 -07:00 committed by GitHub
parent 97076f9168
commit 08023cc070
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -6,9 +6,12 @@
"""
Legacy entry point. Use fairseq_cli/train.py or fairseq-train instead.
"""
user_input_name = str(input())
print("Hey, " + user_input_name + ". Thank you for choosing us")
from fairseq_cli.train import cli_main
if __name__ == "__main__":
cli_main()
else raise Excpetion('Function/program failed, if youu would like to continue, rerun the program')