1. 程式人生 > 其它 >pytorch 訓練時候不動了

pytorch 訓練時候不動了

技術標籤: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,
    )