使用命令行设置超参数
ArgumentParser
简单的可以使用python 工具ArgumentParser。
from argparse import ArgumentParser
parser = ArgumentParser()
# 训练使用的设备参数
parser.add_argument("--devices", type=int, default=2)
# 模型内的超参数
parser.add_argument("--layer_1_dim", type=int, default=128)
# Parse the user inputs and defaults (returns a argparse.Namespace)
args = parser.parse_args()
# 解析后使用相应的参数
trainer = Trainer(devices=args.devices)
model = MyModel(layer_1_dim=args.layer_1_dim)
在终端中命令中设置相应的参数:
python trainer.py --layer_1_dim 64 --devices 1
CLI
如果参数过多,则推荐使用 Lightning CLI。