Skip to content
This repository has been archived by the owner on Jul 5, 2021. It is now read-only.

Error for Crop Height and Width of the FC-DenseNet103? #230

Open
AI-ML-Enthusiast opened this issue Feb 12, 2020 · 0 comments
Open

Error for Crop Height and Width of the FC-DenseNet103? #230

AI-ML-Enthusiast opened this issue Feb 12, 2020 · 0 comments

Comments

@AI-ML-Enthusiast
Copy link

@GeorgeSeif
Thanks for your awesome repository. I am training FCDenseNet-103 on my own dataset. It is fine with Crop_height and width=512, but when I change it to 1024, it gives the following error:

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[1,288,256,256] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[node FC-DenseNet103_3/denseblock3/BatchNorm_6/FusedBatchNorm (defined at D:\feasibilityAnalysis\RiversideProject\Code\Semantic-SegmentationTF\models\FC_DenseNet_Tiramisu.py:13) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

[[node Mean_3 (defined at D:/feasibilityAnalysis/RiversideProject/Code/Semantic-SegmentationTF/train.py:102) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

My environment:
Windows 10
Tensorflow 1.13
python3.6
GPU

Is there any clue to solve this issue?

Thanks in Advanced

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant