if __name__ == "__main__":
pip_install_requirements()
setup()
After Change
success, errmsg = pip_install_requirements()
if not success:
print("Failed to install torchbenchmark requirements:")
print(errmsg)
if not args.continue_on_fail:
sys.exit(-1)
success &= setup(verbose=args.verbose, continue_on_fail=args.continue_on_fail)
