lschirmer/Attention-Augmented-Convolutional-Keras-Networks

Implementation of Attention Augmented Convolutional Networks using Keras

Python

Issues

I am using AttentionAugConv2D layer form "https://github.com/lschirmer/Attention-Augmented-Convolutional-Keras-Network" link in my code. I am working on 256x256x3 size of images. whenever I start training, I get "OOM when allocating tensor with shape[2,2,256,256,256,256] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator cpu" issue
#1 opened 2 years ago
2