ValueError: You selected an invalid strategy name: `strategy=None`. It must be either a string or an instance of `pytorch_lightning.strategies.Strategy`. Example choices: auto, ddp, ddp_spawn, deepspeed, ... Find a complete list of options in our documentation at https://lightning.ai
Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
你遇到的错误与PyTorch Lightning的分布式训练策略有关。看起来你已经指定了 strategy=None
,这不是一个有效的选项。在使用PyTorch Lightning时,你需要选择一个有效的分布式训练策略。
错误消息建议一些示例,如 auto
、ddp
、ddp_spawn
和 deepspeed
。你可以在PyTorch Lightning文档中找到完整的选项列表。
以下是解决方法:
# 导入必要的库
import pytorch_lightning as pl
# 选择分布式训练策略
strategy = "auto" # 替换为你想使用的策略,如 "auto"、"ddp"、"ddp_spawn" 等
# 使用选择的策略初始化训练器
trainer = pl.Trainer(strategy=strategy)
# 现在你可以使用训练器来训练你的LightningModule
具体的例子,比如我的代码,如下: