我尝试在从编码器获得的输出上使用layernorm，其尺寸为（1，497，64，64），即（batch_size，channels，height，width）。我尝试规范化此输入，以便进一步将其提供给流模型。在应用layernorm之前，没有错误，但在应用之后，我得到以下错误：“运行时错误：cuda运行时错误（77）：在/tmp/pip-req-build-irc 6 u2 ci/aten/src/THC/THCReduceAll.cuh：327处遇到非法内存访问”
yzxexxkh1#
这听起来像是你在将Layernorm应用到编码器的输出后遇到了一个错误。如果没有更多的上下文，很难诊断这个问题。这里有两件事你可以检查：
(batch_size, channels, height, width)
channels
1条答案
按热度按时间yzxexxkh1#
这听起来像是你在将Layernorm应用到编码器的输出后遇到了一个错误。如果没有更多的上下文，很难诊断这个问题。
这里有两件事你可以检查：
(batch_size, channels, height, width)。请确保channels维度与数据中的要素数量匹配。