我正在尝试用Keras在google Colab上实现一个自我关注的GAN。当我测试我的注意力层时,我得到了一个OOM错误。那么,我是不是在矩阵乘法上做错了什么,或者只是在更高分辨率(> 64 x 64)的colab GPU上进行了一次太昂贵的操作?attention_o.attentionMap(tensor) 这是错误: OOM when allocating tensor with shape[32,4096,4096] and type double 非常感谢您的关注
我收到以下错误: RuntimeError: mat1 and mat2 shapes cannot be multiplied (32x33856 and 640000x256) 我不明白我需要如何改变我的网络参数我使用在this paper中创建的网络,并尝试修改参数以满足我的needs.This是代码,我更改了第一个卷积的参数,但仍然得到错误: class ChordClassificationNetwork