2021-02-04から1日間の記事一覧
まとめ model.cuda()をしてTrainしようと思ったらconvのパラメータがcudaに入ってないと言われる いろいろ試してみた所、model.cuda()をした後にレイヤーを変えたりしたのが原因だった 全部終わった後にmodel.cuda()をすることで対応 コードサンプル エラー…
まとめ model.cuda()をしてTrainしようと思ったらconvのパラメータがcudaに入ってないと言われる いろいろ試してみた所、model.cuda()をした後にレイヤーを変えたりしたのが原因だった 全部終わった後にmodel.cuda()をすることで対応 コードサンプル エラー…