Lenet 5 with cpp implementation and cuda optimazation in convolutional layers.
- C/C++
- Bazel
- Eigen
- Cuda
# evaluate
bazel run //:evaluate # cpu
bazel run //:evaluate --config=cuda --//:conv_ver=v1 # cuda, using conv v1
bazel run //:evaluate --config=cuda --//:conv_ver=v2 # cuda, using conv v2
bazel run //:evaluate --config=cuda --//:conv_ver=v3 # cuda, using conv v3 (default)
# test
bazel run //test:test --config=cuda --//:conv_ver=<ver>
There are 3 versions of cuda optimization:
..... so lazy to write ๐ด .....