PyTorch使用tensorboard可以顯示網絡運行情況,但偶爾使用SummaryWriter時遇到Segmentation fault錯誤。
利用python3的faulthandler,可定位到出錯的代碼行,具體操作有兩種方式如下:
(1) 在代碼中寫入faulthandler
import faulthandler
# 在import之后直接添加以下啟用代碼即可
faulthandler.enable()
# 后邊正常寫你的代碼
(2)直接通過命令行來啟用,運行時添加-X faulthandler參數即可:
python -X faulthandler your_script.py
錯誤代碼:
import xxx
from torch.utils.tensorboard import SummaryWriter
import yyylog_dir =