Skip to content

Instantly share code, notes, and snippets.

View phughk's full-sized avatar

Przemyslaw Hugh Kaznowski phughk

View GitHub Profile
@phughk
phughk / burn-rs-modules.md
Created September 22, 2025 04:23
Brief description of Burn-rs modules

Convolutional Modules

Conv1d / Conv2d / Conv3d

Apply a sliding filter to input signal. Used for spatial data (ex. 1d=audio, 2d=image, 3d=video)

ConvTranspose1d / ConvTranspose2d / ConvTranspose3d

Also known as deconvolution, they perform the reverse of convolution by upsampling

DeformConv2d

A more advanced convolution that allows for sampling to be offset, making it better for geometric transformations.

Recurrent and Sequential Modules

Gru

Gated Recurrent Unit