DelurGan トレーニング FAQ

Q: AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'

A: これは、CPU のみのプラットフォームで実行している一方で、トレーニングコマンドが CUDA デバイスを選択しようとしていることが原因です。

プロジェクトが CPU モードで実行されるように、コマンドに --gpu_ids -1 を追加します。

python train.py --gpu_ids -1

GPU でトレーニングする想定だった場合は、コマンドを変更する前に、ローカルの PyTorch ビルドに CUDA サポートが含まれていることを確認してください。

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)

Q: ImportError: cannot import name 'SSIM'

A: ssim パッケージから ssim をインポートし、SSIM という別名を付けます。

from ssim import ssim as SSIM

これにより、インストール済みモジュールが関数を ssim として公開している一方で、コード側が SSIM という名前を期待している場合の問題が修正されます。