gaitset程序踩坑

程序来自 https://github.com/AbnerHqC/GaitSet

Pretreatment.py

报错:module 'scipy.misc' has no attribute 'imsave'

问题原因:版本不对 

解决方法:pip install scipy==1.2.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

有可能需要安装 xarray

ValueError: Object arrays cannot be loaded when allow_pickle=False

报错原因:自Numpy 1.16.3版本发行之后,函数 numpy.load() 和 numpy.lib.format.read_array() 采用allow_pickle关键字,现在默认为False以响应CVE-2019-6446 < nvd.nist.gov/vuln/detail / CVE-2019-6446 >。降级到1.16.2对我有帮助,因为错误发生在一些library内

解决方案:将Numpy版本降级到1.16.2及以下。

操作步骤如下:

使用conda的用户,输入指令:conda install numpy==1.16.2

使用pip的用户,输入指令:pip install numpy==1.16.2

 

train.py

pytorch container.py... IndexError: index 0 is out of range

重载了模型,换一个工作目录  //    尝试只用一个gpu

pytorch RuntimeError: CUDA error: out of memory

nvidia-smi 查看gpu占用情况

你可能感兴趣的:(神经网络相关)