Skip to content

Commit

Permalink
examples: cnn_mnist_dag.m: fix CPU-variant
Browse files Browse the repository at this point in the history
  • Loading branch information
vedaldi committed Sep 20, 2015
1 parent bc09a4c commit c7768a3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/cnn_mnist_dag.m
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@
% --------------------------------------------------------------------
images = imdb.images.data(:,:,:,batch) ;
labels = imdb.images.labels(1,batch) ;
if numel(opts.useGpu) > 0
if opts.useGpu > 0
images = gpuArray(images) ;
end
inputs = {'input', images, 'label', labels} ;
Expand Down

0 comments on commit c7768a3

Please sign in to comment.