pytorch lightning之使用命令行设置超参数

使用命令行设置超参数

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。