問題タブ [amd-rocm]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
23 問題
0
投票する
0
に答える
67
参照
pytorch - ROCM/MIOpen を使用した PyTorch での 3D 畳み込みの問題 - パラメータが正しくないため、Forward Convolution を実行できません
AMD GPU を搭載したシステムで、畳み込みオートエンコーダーに PyTorch の nn.Conv3d を使用しようとしています。最新の ROCM (4.5) と MIOpen (2.14) があります。同じトレーニング スクリプトが NVIDIA GPU でも機能します。nn.Conv2D で同じトレーニングを取得できましたが、Conv3D の場合、次のエラーが発生します。
ここにネットワークがあります：
トレーニングループは次のとおりです。
何か案は？