pytorch逐漸吸引了更多的學(xué)術(shù)界以及工業(yè)界的注目,很多人在用pytorch時(shí)依然放不下tensorflow的可視化神奇tensorboard,如何能在使用
pytorch的同時(shí)享受tensorboard給我們帶來的視覺體驗(yàn)?zāi)赜苌В看鸢妇褪莟ensorboardX搀庶,這里強(qiáng)力安利一波:
先放github鏈接:?tenserboardX_github
安裝極其簡便:
pip install tensorboardX
或者:
pip install git+https://github.com/lanpa/tensorboard-pytorch
然后就用吧哈哈~~
運(yùn)行自己寫的train.py會(huì)在“./log”目錄下生成一個(gè)events.out.tfevents.{time}.{machine-name}的文件闰蚕,需要注意的是不要用中文命名目錄,中文目錄中看不到任何圖形司草。然后還有一個(gè)問題是我們跑程序時(shí)候往往會(huì)各種調(diào)參各種換模型的跑,會(huì)在一個(gè)log下面生成大量這種文件泡仗,而tensorboard是沒法讀取超過兩個(gè)的(貌似)翻伺,所以我的解決方案是這樣:
要看的留在log文件夾下,把不用的放到另一個(gè)文件夾就行了沮焕,簡單粗暴
當(dāng)然如果是在公司使用吨岭,可能會(huì)有跳板機(jī),這樣在用的時(shí)候就需要注意了峦树,直接用給的網(wǎng)址就會(huì)找不到了辣辫,那么該怎么做呢?
其實(shí)也很簡單魁巩,一般跳板機(jī)只開放了某幾個(gè)端口急灭,必須從里面挑選使用,所以tensorboardX默認(rèn)的6006端口或者visdom(facebook自己出的可視化工具)默認(rèn)的8097端口就用不了了谷遂,需要制定端口才行慕的,跳板機(jī)會(huì)有端口映射信息,比如這樣:
比如我們指定8003端口:
tensorboard --logdir='./log' --port=8003
這樣在網(wǎng)址上輸入8003對(duì)應(yīng)的ip:port就可以啦弹沽,比如我這里就是 http://mesos-gpu-online020-bjdx11.cloud.qiyi.domain:31006脆荷,不用理會(huì)它讓我們輸?shù)木W(wǎng)址。其中l(wèi)ogdir填寫存放生成文件的地址集晚,一般是python文件當(dāng)前目錄下的logs文件夾窗悯。
如果出現(xiàn)以下情況:
很可能是logdir寫錯(cuò)了,注意一下絕對(duì)路徑相對(duì)路徑的問題偷拔,應(yīng)該就沒錯(cuò)了~
visdom就很類似了
python -m visdom.server -port=8003