pytorch训练ImageNet笔记(三)

1.训练过程中有可能会报错:

/storage/st8/public/python3/lib/python3.5/site-packages/PIL/TiffImagePlugin.py:747: UserWarning: Possibly corrupt EXIF data.  Expecting to read 1835008 bytes but only got 0. Skipping tag 0
  " Skipping tag %s" % (size, len(data), tag))

这是因为数据集中的某些图片有可能遭到了破坏,删除相关图片即可

import piexif
piexif.remove(img)

2.刚开始训练的时候,没有接触过argparse包,该包是用来设定参数的,使用方法如下:

猜你喜欢

转载自blog.csdn.net/wu_x_j_/article/details/84642812