apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix failing tests.

Browse Source
This commit is contained in:
akhandait
2019-10-10 13:44:53 +05:30
parent 61fcdeb221
commit d2c78df037
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/transposed_convolution_impl.hpp
+5 -7
View File
@@ -78,15 +78,15 @@ TransposedConvolution<
{
weights.set_size((outSize * inSize * kW * kH) + outSize, 1);
// TODO: Use the Padding layer.
// padding = new Padding<>(padW, padW, padH, padH);
// padding = new Padding<>(this->padW, this->padW, this->padH, this->padH);
aW = (outputWidth + kW - 2 * padW - 2) % dW;
aH = (outputHeight + kH - 2 * padH - 2) % dH;
aW = (outputWidth + kW - 2 * this->padW - 2) % dW;
aH = (outputHeight + kH - 2 * this->padH - 2) % dH;
// Check if the output height and width are possible given the other
// parameters of the layer.
if (outputWidth != dW * (inputWidth - 1) + aW + 2 * padW + 2 - kW ||
outputHeight != dH * (inputHeight - 1) + aW + 2 * padH + 2 - kH)
if (outputWidth != dW * (inputWidth - 1) + aW + 2 * this->padW + 2 - kW ||
outputHeight != dH * (inputHeight - 1) + aH + 2 * this->padH + 2 - kH)
{
Log::Fatal << "The output width / output height is not possible given "
<< "the other parameters of the layer." << std::endl;
@@ -340,8 +340,6 @@ void TransposedConvolution<
ar & BOOST_SERIALIZATION_NVP(inputHeight);
ar & BOOST_SERIALIZATION_NVP(outputWidth);
ar & BOOST_SERIALIZATION_NVP(outputHeight);
ar & BOOST_SERIALIZATION_NVP(aW);
ar & BOOST_SERIALIZATION_NVP(aH);
if (Archive::is_loading::value)
{