search

ikhlestov / vision_networks

Repo about neural networks for images handling
MIT License
264 stars 122 forks source link

Is there any memory-efficient tensorflow Implementation？ #8

Closed feynman233 closed 7 years ago

feynman233 commented 7 years ago

Hi, I have found the same problem about the GPU memory，and is there any memory-efficient tensorflow Implementation？ Thanks very much!

ikhlestov commented 7 years ago

Hi! I think there should be some memory-efficient implementations, but I don't know exactly any repo. And in this one no memory efficient implementation and at this moment I don't intend to do this. But you are welcome to send pull request)

tengshaofeng commented 7 years ago

hi,@feynman233 @ikhlestov dose it mean that it can not train densenet_BC on 224224 pictures? my gpu is 12GB, and i try to train on 224224 images ,set the batchsize to 10, it comes the error 'OOM when allocating tensor with shape[10,224,224,108]'

ikhlestov commented 7 years ago

Hi! It seems so. Unfortunately I have no access to the ImageNet dataset, so I not test the network on it.