【Pytorch】問題解決:No module named 'ranger'

前言

這篇是我在使用 Pytorch 的時候碰到的問題,出現了以下的訊息:

No module named 'ranger'

解決方法

Ranger 在此指的是 Deep Learning 的一種 Optimizer,
由於目前還不能夠透過 PyPI 安裝,
只能手動去官方的 github 安裝。

這邊就不多介紹這種 Ranger Optimizer 的細節了,著重在解決問題

去官方的 github:

依照官方說的 github,進行以下的安裝指令:

git clone https://github.com/lessw2020/Ranger-Deep-Learning-Optimizer
cd Ranger-Deep-Learning-Optimizer
pip install -e . 

使用方式 (Ranger 2021)

from ranger21 import Ranger21

optimizer = Ranger21(model.parameters(), lr=learning_rate, num_epochs=num_epochs, num_batches_per_epoch = BATCH_SIZE)