一般是

python train.py --bath_size 5

利用argparse解析參數

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('integer', type=int, help='display an integer')
args = parser.parse_args()

參數類型

可選參數

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--integer', type=int, help='display an integer')
args = parser.parse_args()

使用

python train.py --integer 5

定位參數

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('integer', type=int, help='display an integer')
args = parser.parse_args()

在用的時候不用輸入integer直接輸入數字就行了

python train.py   5