2018-10-13から1日間の記事一覧

PyTorchのDataParallelのモデルを保存する

PyTorchで複数GPUで学習させる場合， model = nn.DataParallel(model, device_ids=[0,1,2]) のようにDataParallelで保存しますが，このモデルを保存したい場合にcuda runtime error : out of memoryが出ることがあります． その場合は，下のようにDataParall…