mlpack features I am working on.
Contains all the work done on the Spatial Dropout layer. This layer does an approximate inverse of the Max Pool operation.
MaxPooling Operation and Corresponding UnPool operation.
UnPool.ipynb
- Understanding the working of the PyTorch layer.test.cpp
- Armadillo implementation of the layer’s functionality.maxpool.txt
- The output obtained from running the cpp file.To run the .cpp
file, you need to have armadillo installed in your system.
g++ test.cpp -o test -larmadillo && ./test > maxpool.txt
To run UnPool.ipynb
, you would need the pytorch
python library.