PyTorch使用tensorboard的SummaryWriter報(bào)錯(cuò)問題解決方案
PyTorch使用tensorboard可以顯示網(wǎng)絡(luò)運(yùn)行情況,但偶爾使用SummaryWriter時(shí)遇到Segmentation fault錯(cuò)誤。
利用python3的faulthandler,可定位到出錯(cuò)的代碼行,具體操作有兩種方式如下:
(1) 在代碼中寫入faulthandler
import faulthandler # 在import之后直接添加以下啟用代碼即可 faulthandler.enable() # 后邊正常寫你的代碼
(2)直接通過命令行來啟用,運(yùn)行時(shí)添加-X faulthandler參數(shù)即可:
python -X faulthandler your_script.py
錯(cuò)誤代碼:
import xxx from torch.utils.tensorboard import SummaryWriter import yyy log_dir = './run_logs' self.writer = SummaryWriter(log_dir)
報(bào)錯(cuò)如下:
File "/opt/conda/lib/python3.7/site-packages/tensorboard/compat/__init__.py", line 45 in tf
File "/opt/conda/lib/python3.7/site-packages/tensorboard/lazy.py", line 50 in load_once
File "/opt/conda/lib/python3.7/site-packages/tensorboard/lazy.py", line 97 in wrapper
File "/opt/conda/lib/python3.7/site-packages/tensorboard/lazy.py", line 65 in __getattr__
File "/opt/conda/lib/python3.7/site-packages/tensorboard/summary/writer/event_file_writer.py", line 72 in __init__
File "/opt/conda/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py",line 66 in __init__
File "/opt/conda/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py",line 256 in _get_file_writer
File "/opt/conda/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py", line 225 in __init__
經(jīng)過查詢,發(fā)現(xiàn)極有可能是導(dǎo)包(from torch.utils.tensorboard import SummaryWriter)順序?qū)е碌膯栴}。參考鏈接:[TensorBoard] The different order of import SummaryWriter may cause Segmentation fault
將from torch.utils.tensorboard import SummaryWriter順序調(diào)前,可以順利導(dǎo)包和實(shí)例化了。
import xxx from torch.utils.tensorboard import SummaryWriter import yyy
改為
from torch.utils.tensorboard import SummaryWriter import xxx import yyy
到此這篇關(guān)于PyTorch使用tensorboard的SummaryWriter報(bào)錯(cuò)的文章就介紹到這了,更多相關(guān)PyTorch SummaryWriter報(bào)錯(cuò)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
- Pytorch?Conda環(huán)境pack打包遷移報(bào)錯(cuò)的處理方案
- Pytorch backward報(bào)錯(cuò)2次訪問計(jì)算圖需要retain_graph=True的情況詳解
- 分析PyTorch?Dataloader報(bào)錯(cuò)ValueError:num_samples的另一種可能原因
- pytorch報(bào)錯(cuò)問題:ValueError: num_samples should be a positive integer value, but got num_samples=0
- 解決遇到:PytorchStreamReader failed reading zip archive:failed finding central錯(cuò)誤問題
相關(guān)文章
原來我一直安裝 Python 庫的姿勢(shì)都不對(duì)呀
平常我都是直接執(zhí)行 pip install 安裝的第三方庫,很多教程也是這么介紹的,一直以來我都認(rèn)為這是標(biāo)準(zhǔn)的、正確的安裝 Python 第三方庫的姿勢(shì)。下面小編給大家分享一篇教程,一起看看吧2019-11-11
Python實(shí)現(xiàn)一個(gè)簡單三層神經(jīng)網(wǎng)絡(luò)的搭建及測(cè)試 代碼解析
一個(gè)完整的神經(jīng)網(wǎng)絡(luò)一般由三層構(gòu)成:輸入層,隱藏層(可以有多層)和輸出層。本文所構(gòu)建的神經(jīng)網(wǎng)絡(luò)隱藏層只有一層。一個(gè)神經(jīng)網(wǎng)絡(luò)主要由三部分構(gòu)成(代碼結(jié)構(gòu)上):初始化,訓(xùn)練,和預(yù)測(cè)。,需要的朋友可以參考下面文章內(nèi)容的具體內(nèi)容2021-09-09
淺談Python2之漢字編碼為unicode的問題(即類似\xc3\xa4)
今天小編就為大家分享一篇淺談Python2之漢字編碼為unicode的問題(即類似\xc3\xa4),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-08-08
python?包之?multiprocessing?多進(jìn)程
這篇文章主要介紹了python?包之?multiprocessing?多進(jìn)程,首先創(chuàng)建一個(gè)進(jìn)程。實(shí)例化?Process?類創(chuàng)建一個(gè)進(jìn)程對(duì)象、然后調(diào)用它的?start?方法即可生成一個(gè)子進(jìn)程,下文更多相關(guān)資料,需要的朋友可以參考一下2022-04-04
對(duì)tensorflow中tf.nn.conv1d和layers.conv1d的區(qū)別詳解
今天小編就為大家分享一篇對(duì)tensorflow中tf.nn.conv1d和layers.conv1d的區(qū)別詳解,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-02-02

