mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Make conv2d work again
This commit is contained in:
@@ -5,19 +5,20 @@
|
||||
#include <vector>
|
||||
|
||||
#include "activations.cuh"
|
||||
#include "padding.cuh"
|
||||
|
||||
namespace Layers {
|
||||
|
||||
class Conv2d {
|
||||
public:
|
||||
Conv2d(
|
||||
int inputSize,
|
||||
int inputChannels,
|
||||
int kernelSize,
|
||||
int stride,
|
||||
std::string padding,
|
||||
int numFilters,
|
||||
Activation activation
|
||||
int inputSize,
|
||||
int inputChannels,
|
||||
int kernelSize,
|
||||
int stride,
|
||||
Padding padding,
|
||||
int numFilters,
|
||||
Activation activation
|
||||
);
|
||||
~Conv2d();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user