mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Rework padding size setting
This commit is contained in:
@@ -24,10 +24,12 @@ class ModelTest : public ::testing::Test {
|
||||
CUDANet::Model *model =
|
||||
new CUDANet::Model(inputSize, inputChannels, outputSize);
|
||||
|
||||
int paddingSize = 0;
|
||||
|
||||
// Conv2d
|
||||
CUDANet::Layers::Conv2d *conv2d = new CUDANet::Layers::Conv2d(
|
||||
inputSize, inputChannels, kernelSize, stride, numFilters,
|
||||
CUDANet::Layers::Padding::VALID,
|
||||
paddingSize,
|
||||
CUDANet::Layers::ActivationType::NONE
|
||||
);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user