Parse input arguments
()
| 828 | |
| 829 | |
| 830 | def parse_args(): |
| 831 | """ |
| 832 | Parse input arguments |
| 833 | """ |
| 834 | parser = argparse.ArgumentParser(description='run Sample Code Test') |
| 835 | parser.add_argument('--debug', dest='debug', action="store_true") |
| 836 | parser.add_argument('--full-test', dest='full_test', action="store_true") |
| 837 | parser.add_argument( |
| 838 | '--mode', dest='mode', type=str, default='cpu', help='run on device' |
| 839 | ) |
| 840 | parser.add_argument( |
| 841 | '--build-doc', |
| 842 | dest='build_doc', |
| 843 | action='store_true', |
| 844 | help='build doc if need.', |
| 845 | ) |
| 846 | parser.add_argument( |
| 847 | '--gpu_id', |
| 848 | dest='gpu_id', |
| 849 | type=int, |
| 850 | default=0, |
| 851 | help='GPU device id to use [0]', |
| 852 | ) |
| 853 | parser.add_argument( |
| 854 | '--logf', dest='logf', type=str, default=None, help='file for logging' |
| 855 | ) |
| 856 | parser.add_argument( |
| 857 | '--threads', |
| 858 | dest='threads', |
| 859 | type=int, |
| 860 | default=0, |
| 861 | help='sub processes number', |
| 862 | ) |
| 863 | |
| 864 | args = parser.parse_args() |
| 865 | return args |
no test coverage detected