Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add model predict test

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-22 22:31:32 +01:00
parent 9ecd51c63a
commit 9482d7bc43
5 changed files with 145 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/layers/max_pooling.cuh
View File

@@ -1,8 +1,6 @@
#ifndef CUDANET_MAX_POOLING_H
#define CUDANET_MAX_POOLING_H
#include <cuda_runtime.h>
#include "activation.cuh"
#include "layer.cuh"