Jump to content
Compvision.ru
vitaly-post

Ошибка ResourceExhaustedError

Recommended Posts

При обучении Mask R-CNN питон вываливается с ошибкой:

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[2,256,256,256] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

Вероятно нехватает памяти у видеокарты (nVIDIA GeForce GTX 670   2Гб)

Кто знает, какую видеокарту нужно, чтобы хоть как-то работала NN на распознавание изображений?

Share this post


Link to post
Share on other sites

Хоть как-то на распознавании (inference) или хоть как-то на обучении (train)? Для обучения лучше 8 Гб памяти и больше.

Share this post


Link to post
Share on other sites

Create an account or sign in to comment

You need to be a member in order to leave a comment

Create an account

Sign up for a new account in our community. It's easy!

Register a new account

Sign in

Already have an account? Sign in here.

Sign In Now


  • Recently Browsing   0 members

    No registered users viewing this page.

×