CLI 第二步：接上已训练的模型

Writer 读取 best.pt；write 接收开头、关键词和行长；最后把四行拼起来显示。

lessons/24/build_cli/02_write.py · 第 18—24 行（当前源码节选）

 18  writer = Writer(DEFAULT_RUN / "best.pt", device=args.device)
 19  poems = writer.write(
 20      start=args.start,
 21      keywords=parse_keywords(args.keywords),
 22      form=5 if args.form == "five" else 7,
 23  )
 24  print("\n".join(poems[0]["lines"]))

实际执行的命令

.venv/bin/python lessons/24/build_cli/02_write.py --start '春' --device cpu

标准输出

春色三分别，
春来一日来。
东风吹一笑，
莫负酒家开。

执行结果

退出码 0 · 2026-09-19T00:55:52.173059+00:00