Image_manipulation_detection icon indicating copy to clipboard operation
Image_manipulation_detection copied to clipboard

UserWarning: Converting sparse IndexedSlices to a dense Tensor of unknown shape.

Open yukiiiiihan opened this issue 6 years ago • 0 comments

Hi @LarryJiang134 ,when I trained the model with train.py,I received this warning: UserWarning: Converting sparse IndexedSlices to a dense Tensor of unknown shape. This may consume a large amount of memory. "Converting sparse IndexedSlices to a dense Tensor of unknown shape. "

and this mistake: Allocator (GPU_0_bfc) ran out of memory trying to allocate 3.49GiB. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available. However,my GPU has 8G memory and 6.5G of free memory. Is it true that the program requires 8G+ of memory? Is it possible to train on the basis of not modifying the network?

yukiiiiihan avatar Jan 07 '19 08:01 yukiiiiihan