lschirmer Attention-Augmented-Convolutional-Keras-Networks issues - Githubissues

lschirmer / Attention-Augmented-Convolutional-Keras-Networks

Implementation of Attention Augmented Convolutional Networks using Keras

8 stars 3 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

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 patilparam-edgeneural opened 2 years ago
2