训练前准备
- 为保证后续训练能够顺利执行,可先修改训练脚本(main.py)中以下代码(以下行数仅为参考,请以实际所在行数为准)。
- 检查numpy和pillow版本,需要numpy版本大于等于1.23.0,pillow版本大于等于8.2.0,若不满足需根据使用的Python版本进行如下操作。
- 若用户使用Python3.8及以上版本,可执行如下命令升级相关依赖。
pip3 install numpy==1.23.0 pip3 install pillow==8.2.0
- 若用户使用Python3.7.x(3.7.0~3.7.11)版本,需注释掉x2ms_adapter目录下__init__.py文件中部分代码(以下行数仅为参考,请以实际所在行数为准)。
图1 注释部分代码
- 若用户使用Python3.8及以上版本,可执行如下命令升级相关依赖。
父主题: 分析迁移(X2MindSpore)