在910A上拉起 stylegan2-ada-pytorch 训练中，遇到zipfile.BadZipFile: File is not a zip file

一、问题现象（附报错日志上下文）：
部署项目，硬件设备Atlas 800 9000，910PremiumA，使用bash拉起单卡的遇到数据集报错
modelzoo仓代码链接：https://gitee.com/ascend/ModelZoo-PyTorch/tree/master/PyTorch/contrib/cv/others/stylegan2-ada-pytorch

二、软件版本:
-- CANN 版本 8.1
-- Python 版本 3.7.12
-- 操作系统版本 openeuler22.03

三、测试步骤：
根据对应文件中的requestment安装对应环境后，拉起单卡训练
启动命令为：bash train_full_1p.sh --data_path==/home/test/ModelZoo-PyTorch/PyTorch/contrib/cv/others/stylegan2-ada-pytorch/data/Animal_Faces_datasets.zip ，其中data为官网下载的afhq数据集
![输入图片说明](https://foruda.gitee.com/images/1760689601747491320/f6221304_8997601.png "屏幕截图")
查看output输出日志信息，发现报错为data-path为否zip文件，通过ls -la去查看数据集文件，同时解压后，发现zip文件也没有损坏
![输入图片说明](https://foruda.gitee.com/images/1760689736535280377/287c3bb1_8997601.png "屏幕截图")
![输入图片说明](https://foruda.gitee.com/images/1760689800503497585/ff3c0744_8997601.png "屏幕截图")

日志报错信息：
(gan) [root@bms-jinc test]# cat output/0/train_0.log 
[W OperatorEntry.cpp:121] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::_has_compatible_shallow_copy_type(Tensor self, Tensor from) -> (bool)
    registered at /usr1/v1.8.1/pytorch/build/aten/src/ATen/RegisterSchema.cpp:20
  dispatch key: Math
  previous kernel: registered at /usr1/v1.8.1/pytorch/build/aten/src/ATen/RegisterMath.cpp:5686
       new kernel: registered at /usr1/workspace/FPTA_Daily_Plugin_open_date/Plugin/torch_npu/csrc/aten/ops/HasCompatibleShallowCopyType.cpp:37 (function registerKernel)
Traceback (most recent call last):
  File "train.py", line 558, in <module>
    main() # pylint: disable=no-value-for-parameter
  File "/root/miniforge3/envs/gan/lib/python3.7/site-packages/click/core.py", line 1161, in __call__
    return self.main(*args, **kwargs)
  File "/root/miniforge3/envs/gan/lib/python3.7/site-packages/click/core.py", line 1082, in main
    rv = self.invoke(ctx)
  File "/root/miniforge3/envs/gan/lib/python3.7/site-packages/click/core.py", line 1443, in invoke
    return ctx.invoke(self.callback, **ctx.params)
  File "/root/miniforge3/envs/gan/lib/python3.7/site-packages/click/core.py", line 788, in invoke
    return __callback(*args, **kwargs)
  File "/root/miniforge3/envs/gan/lib/python3.7/site-packages/click/decorators.py", line 33, in new_func
    return f(get_current_context(), *args, **kwargs)
  File "train.py", line 506, in main
    run_desc, args = setup_training_loop_kwargs(**config_kwargs)
  File "train.py", line 127, in setup_training_loop_kwargs
    training_set = dnnlib.util.construct_class_by_name(**args.training_set_kwargs) # subclass of training.dataset.Dataset
  File "/home/test/ModelZoo-PyTorch/PyTorch/contrib/cv/others/stylegan2-ada-pytorch/dnnlib/util.py", line 289, in construct_class_by_name
    return call_func_by_name(*args, func_name=class_name, **kwargs)
  File "/home/test/ModelZoo-PyTorch/PyTorch/contrib/cv/others/stylegan2-ada-pytorch/dnnlib/util.py", line 284, in call_func_by_name
    return func_obj(*args, **kwargs)
  File "/home/test/ModelZoo-PyTorch/PyTorch/contrib/cv/others/stylegan2-ada-pytorch/training/dataset.py", line 168, in __init__
    self._all_fnames = set(self._get_zipfile().namelist())
  File "/home/test/ModelZoo-PyTorch/PyTorch/contrib/cv/others/stylegan2-ada-pytorch/training/dataset.py", line 190, in _get_zipfile
    self._zipfile = zipfile.ZipFile(self._path)
  File "/root/miniforge3/envs/gan/lib/python3.7/zipfile.py", line 1258, in __init__
    self._RealGetContents()
  File "/root/miniforge3/envs/gan/lib/python3.7/zipfile.py", line 1325, in _RealGetContents
    raise BadZipFile("File is not a zip file")
zipfile.BadZipFile: File is not a zip file
/root/miniforge3/envs/gan/lib/python3.7/site-packages/torch/utils/cpp_extension.py:3: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
  import imp
/root/miniforge3/envs/gan/lib/python3.7/site-packages/torch/utils/cpp_extension.py:23: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
  from pkg_resources import packaging  # type: ignore

Ascend/ModelZoo-PyTorch

内容风险标识

评论 (0)

Ascend/ModelZoo-PyTorch .gitee-modal { width: 500px !important; }

内容风险标识