训练前准备

为保证后续训练能够顺利执行，可先修改训练脚本（main.py）中以下代码（以下行数仅为参考，请以实际所在行数为准）。

1. 将以下两处数据集路径改为数据集实际存放路径。

   

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