Out of Memory Error - Githubissues

xw-hu / DSC

Direction-Aware Spatial Context Features for Shadow Detection and Removal | CVPR 2018 (Oral) & TPAMI 2019

141 stars 35 forks source link

Out of Memory Error #5

Closed msiraj83 closed 6 years ago

msiraj83 commented 6 years ago

Hi I am using your code and want to retrain the model on my own data. but i got out of memory error. Following is the error and GPU specification. Can you you tell me how much memory is required for training ? dsc_training_error

GPU Specification nvidia_specification

msiraj83 commented 6 years ago

Solve the problem by changing the parameter new_height: 400 new_width: 400 to 350 each in example/DSC_detection/DSC.protobox file.