昇腾社区首页
中文
注册

训练前准备

  • 为保证后续训练能够顺利执行,可先修改训练脚本(main.py)中以下代码(以下行数仅为参考,请以实际所在行数为准)。
    1. 将以下两处数据集路径改为数据集实际存放路径。

    2. 将第1处代码取消注释并注释第2处代码。

  • 检查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 注释部分代码