如何在aiForge中使用TensorBoard?

Image for post
Image for post
  1. 如圖所示,如果要使用aiForge的話,必須點擊進入TensorBoard才能正常顯示視覺化的Log資料
  2. 除此之外,當我們在進行訓練的時候,請找出程式碼中,有TensorBoard相關API的部份。
  3. 假設當下有4顆GPU進行4種不同的訓練任務,TensorBoard可以個別顯示,在左下角

它的作用原理就是將SummaryWriter寫入的內容從原本目前的資料夾(預設),移動到/root/notebooks/tensorflow/logs當中。

1/10更新:每完成一次的訓練(例如300 epochs)後,請將logs資料夾移除,再重新建立logs資料夾,藉此清空前次訓練得到的紀錄,要保留也可以。

1/28更新:如果要使用soft link的話,可以使用以下指令:

Image for post
Image for post

注意事項

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store