attributeError:模块' torch.distributed'没有属性' is_initialized'在pytorch == 1.11.x中

发布于 2025-02-07 17:09:31 字数 998 浏览 3 评论 0原文

每当我们为训练量的对象创建对象时,

from transformers import Trainer, TrainingArguments
batch_size = 64
logging_steps = len(emotions_encoded["train"]) // batch_size
model_name = f"{model_ckpt}-finetuned-emotion"
training_args = TrainingArguments(output_dir=model_name,
                                      num_train_epochs=2,
                                      learning_rate=2e-5,
                                      per_device_train_batch_size=batch_size,
                                      per_device_eval_batch_size=batch_size,
                                      weight_decay=0.01,
                                      evaluation_strategy="epoch",
                                      disable_tqdm=False,
                                      logging_steps=logging_steps,
                                      push_to_hub=True,
                                      log_level="error")

我都会得到

属性:模块“ torch.distributed”没有属性'is_initialized'

whenever we are creating the object for TrainingArguments

from transformers import Trainer, TrainingArguments
batch_size = 64
logging_steps = len(emotions_encoded["train"]) // batch_size
model_name = f"{model_ckpt}-finetuned-emotion"
training_args = TrainingArguments(output_dir=model_name,
                                      num_train_epochs=2,
                                      learning_rate=2e-5,
                                      per_device_train_batch_size=batch_size,
                                      per_device_eval_batch_size=batch_size,
                                      weight_decay=0.01,
                                      evaluation_strategy="epoch",
                                      disable_tqdm=False,
                                      logging_steps=logging_steps,
                                      push_to_hub=True,
                                      log_level="error")

I am getting

AttributeError: module 'torch.distributed' has no attribute 'is_initialized'

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

雄赳赳气昂昂 2025-02-14 17:09:31

为了解决此问题

实际上,Mac不支持分布式培训设施

,因此此问题正在

解决此问题,

请访问您的Transformers软件包,在我的情况下,您可以

在我的情况下安装它,它是

Desktop/rajesh/pytorch_env/env/lib/python3.8/site-packages/transformers/training_args.py

替换line-1024

如果torch.distributed.is_initialized()和self.local_rank == -1:

如果是true和self.local_rank == -1:

重新启动内核

In order to solve this problem

Actually Window and Mac doesn't support distributed training facility

so this issue is occuring

To solve this problem

go to your transformers package where you install it

in my case it is

Desktop/rajesh/pytorch_env/env/lib/python3.8/site-packages/transformers/training_args.py

replace line-1024
if torch.distributed.is_initialized() and self.local_rank == -1:
with
if True and self.local_rank == -1:

Restart your kernal

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文