96. 模型训练时解决传入参数解析问题

backward.py:

parser = argparse.ArgumentParser()
parser.add_argument("--IMG_H", type=int, default=256)
# args = parser.parse_args()报错
args = parser.parse_known_args()[0]