(标签-conda|关键词-File)

Question

一、问题现象：
Traceback (most recent call last):
File "/root/anaconda3/envs/Hpytorch/models/IEEE_TGRS_LDGnet-main/train_queue.py", line 216, in
pretrained_dict = torch.load('./ViT-B-32.pt', map_location="cpu").state_dict()
File "/root/anaconda3/envs/Hpytorch/lib/python3.9/site-packages/torch_npu/contrib/transfer_to_npu.py", line 67, in decorated
return fn(*args, **kwargs)
File "/root/anaconda3/envs/Hpytorch/lib/python3.9/site-packages/torch_npu/utils/serialization.py", line 122, in load
warnings.warn("'torch.load' received a zip file that looks like a TorchScript archive"
NameError: name 'warnings' is not defined
/root/anaconda3/envs/Hpytorch/lib/python3.9/tempfile.py:831: ResourceWarning: Implicitly cleaning up <TemporaryDirectory '/tmp/tmptkkkcppf'>
_warnings.warn(warn_message, ResourceWarning)

二、软件版本:
-- CANN 版本: 8.0RC1
-- Pytorch 版本: 1.11.0
-- Python 版本: 3.9.16
-- 操作系统版本: Ubuntu 22.04（昇腾910B显卡）

三、测试步骤：
1.在昇腾服务器中进行模型自动迁移
2.在训练脚本中输入import torch_npu和from torch_npu.contrib import transfer_to_npu
3.运行训练脚本出现如上报错
4.检查已经导入warnings

Answer 1

以下回复参考：皆我百晓生、券券喵儿等免费微信小程序作答:

根据给定的错误信息，可以看出在运行训练脚本时出现了一个NameError，提示warnings未定义。为了解决这个问题，你可以尝试在代码中导入warnings模块，然后再次运行训练脚本。

解决方案： 在训练脚本的开头添加以下代码：

import warnings

这样应该可以解决报错问题。如果问题仍然存在，请检查其他部分的代码是否有类似的问题。