search

keras-team / keras-applications

Reference implementations of popular deep learning models.
Other
2k stars 910 forks source link

Dropout error with keras functional API #67

Closed aakashbehl1994 closed 5 years ago

aakashbehl1994 commented 5 years ago

I can't use dropout on keras functional API, I get a resource exhaustion error.

dropout

Before using dropout my code was working fine but now I am getting this error

dropout_error

dropout_error2

taehoonlee commented 5 years ago

@aakashbehl1994, Thank you for trying Keras. The OOM error is related to not Keras but your machine. The dropout layer requires additional memory for the masking matrix even if there is no weights in dropout.