Pytorch
まとめ model.cuda()をしてTrainしようと思ったらconvのパラメータがcudaに入ってないと言われる いろいろ試してみた所、model.cuda()をした後にレイヤーを変えたりしたのが原因だった 全部終わった後にmodel.cuda()をすることで対応 コードサンプル エラー…
個人的に青ざめながらも勉強にはなったのでメモ まとめと教訓 もらったデータの中の特定のデータに不備があったため、訓練中にエラーが出る(device-side assert triggered) DataLoader中のShuffleをFalseにして、問題となっているデータの推定を行う事で解決…
要約 PytorchでLossのbackwardを行うにはLossがスカラーである必要がある。 torch.nn.BCELoss()は通常平均値を与えるためパラメータでreduction='none'と指定することでピクセルごとのLossを得る その後にpixelごとのlayer weightを掛け、torch.mean()として…