已经导入了warnings还是显示NameError: name 'warnings' is not defined
收藏回复举报
已经导入了warnings还是显示NameError: name 'warnings' is not defined
t('forum.solved') 已解决
发表于2024-08-17 16:39:21
0 查看

一、问题现象(附报错日志上下文):

Traceback (most recent call last): 

  File "/root/anaconda3/envs/Hpytorch/models/IEEE_TGRS_LDGnet-main/train_queue.py", line 216, in <module> 

    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

三、测试步骤:
1.在昇腾服务器中进行模型自动迁移

2.在训练脚本中输入import torch_npu和from torch_npu.contrib import transfer_to_npu

3.运行训练脚本出现如上报错

4.检查已经导入warnings

cke_35874.png

cke_38055.png

cke_36878.png

不知道是什么情况

本帖最后由 匿名用户2024/08/21 11:27:13 编辑

我要发帖子