SMART-Lab / smartlearner

SMART-Learner is a machine learning library built with researchers in mind.
BSD 3-Clause "New" or "Revised" License
10 stars 4 forks source link

Add option telling Theano do not transfer a dataset to the GPU #42

Closed MarcCote closed 9 years ago

MarcCote commented 9 years ago

Often you have too much data and it can all fit on the GPU's RAM. If it is your case, you can now provide keep_on_cpu=True as argument to the Dataset constructor. Theano will transfer data as it needs (one batch at the time) to the GPU. See this for more infos https://groups.google.com/forum/#!topic/theano-users/ZagTpkB4hGg