Transpose Convolutional Layer - Githubissues

lhnguyen102 / cuTAGI

CUDA implementation of Tractable Approximate Gaussian Inference

MIT License

24 stars 9 forks source link

Transpose Convolutional Layer #51

Closed lhnguyen102 closed 3 months ago

lhnguyen102 commented 3 months ago

Description

This PR adds the transposed convolutional layer for both CPU and CUDA

Changes Made

Added transposed convolutional layers for both CPU and CUDA
Added python front end for transposed convolutional layers
Added test for autoencoder architecture

Notes for Reviewers

Here is the command to run autoencoder on Mnist dataset

cd pytagi/pytagi_v1
python test_autoencoder.py