pytorch 訓練時候不動了
阿新 • • 發佈:2020-12-24
技術標籤:deeplearning深度學習深度學習pytorch
因為用的執行緒太多, 開啟檔案太多了導致的,如下步驟
1 num_workers變小有助於解決這個問題
train_loader = DataLoader(
dataset=train_set,
batch_size=args.batchsize,
shuffle=True,
num_workers=4,
pin_memory=True,
)
2.提高linux limit
參考https://learnku.com/articles/21457
3.程式中使用了Image.open, 但是沒有close
img = Image.open(imgpath).convert("RGBA")
img.close()
4.不使用多執行緒,使用主執行緒載入資料
train_loader = DataLoader(
dataset=train_set,
batch_size=args.batchsize,
shuffle=True,
num_workers=0,
pin_memory=True,
)