Skip to content

Machine Learning

[PyTorch] 使用 Keras 風格進度條的套件: pkbar

PyTorch is a famous Python deep learning framework

前言

pkbar 是最近在閱讀他人對於 pointer-generator network 實現時意外看到的一種『進度條』實現方法,號稱可以在 PyTorch 上使用 Keras 風格的進度條。 的確,Keras 預設的進度條還滿淺顯易懂,比起需要自己去印出進度條的 PyTorch 來說使用上更加快速。

當然,其實不僅是可以在 PyTorch 框架的程式碼中使用,也可以在任何迴圈中使用。

Read More »[PyTorch] 使用 Keras 風格進度條的套件: pkbar

[Deep Learning] 在 Windows 上搭建 Tensorflow、CUDA、cuDNN 的訓練環境

搭建深度學習環境一直都不是一項簡單的事情,尤其是相當主流的 Nvidia 顯卡和 Tensorflow 的組合,其中版本的問題以及需要額外安裝的驅動、CUDA、cuDNN 就夠讓人頭疼了。而且主流的作業系統是 Linux 而非 Windows ,可以發現教學文明顯就少了些。

Read More »[Deep Learning] 在 Windows 上搭建 Tensorflow、CUDA、cuDNN 的訓練環境

[PyTorch] 如何使用 squeeze() 和 unsqueeze() 函式去改變張量形狀

在剛開始接觸 PyTorch 這個框架時,為了逐漸掌握 PyTorch 搭建模型的方法,我閱讀、執行過許多官方教學文件上的範例程式。那個時候,經常能在範例程式當中見到 squeeze()unsqueeze() 等函式,卻不太明白這兩個函式究竟有什麼樣的用途。

Read More »[PyTorch] 如何使用 squeeze() 和 unsqueeze() 函式去改變張量形狀