Skip to content
Snippets Groups Projects
Commit 6ae9ab41 authored by Pat Alt's avatar Pat Alt
Browse files

added gpu accelaration to lenet

parent 074d6330
No related branches found
No related tags found
1 merge request!7669 initial run including fmnist lenet and new method
......@@ -46,5 +46,5 @@ end
Builds a LeNet-like convolutional neural network.
"""
lenet5(builder=LeNetBuilder(5, 6, 16); kwargs...) = NeuralNetworkClassifier(builder=builder; kwargs...)
lenet5(builder=LeNetBuilder(5, 6, 16); kwargs...) = NeuralNetworkClassifier(builder=builder; acceleration=CUDALibs(), kwargs...)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment