1. 程式人生 > >pytorch訓練ImageNet筆記(三)

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包,該包是用來設定引數的,使用方法如下: