踩坑记录 ValueError: You selected an invalid strategy name: `strategy=None`.

2023-12-14 04:08:29

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

具体的，如下：

文章来源:https://blog.csdn.net/qq_33440910/article/details/134813831
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。