JavaScriptを有効にしてください

PyTorchとメモリ

 ·  ☕ 1 min read
  • 必要なメモリ量

必要メモリ量(byte) =
 (ニューロンの数 × Batch Size + パラメータ数) × 2 (data & grad) × 4(byte)
https://nori-life-log.com/nnの学習で必要なgpuメモリを算出する

  • 重みを固定(freeze)
1
2
3
# freeze all layers
for param in model.parameters():
    param.requires_grad = False
共有

YuWd (Yuiga Wada)
著者
YuWd (Yuiga Wada)
機械学習・競プロ・iOS・Web