Skip to content

frekkoz3/LEARNABLE-POOLER-PROJECT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LEARNABLE POOLER

This project is the final project of the academic course "Introduction to Machine Learning" (year 2024-2025).

The aim of the project is to develop a new type of learnable (parameterized in a way that backpropagation can optimize it) pooler (Pooling Operation, eg MaxPool2d) for standard CNN architecture.

The project is divided in 3 different parts:

  1. Literature review and existing poolers implementation in PyTorch
  2. Implementation of a new pooler in PyTorch
  3. Evaluation of the new pooler on CIFAR10 and MNIST using a standard CNN

The code will be develop in Python files while the notebook is the wrapper for the final product.

More about the project in the notebook.

Author : Francesco Bredariol SM3201379

About

Temperature Driven Gumbel Softmax Pooler

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors