This repository contains the implementation of two distinct neural network architectures aimed at improving existing models:
- A modified version of the baseline model from the Exemplar Diffusion Machine (EDM), specifically the DhariwalUNet. Modifications were made to accommodate hardware constraints.
- An advanced version of the DhariwalUNet that incorporates N3 Blocks from N3Net, replacing the attention mechanisms at the initial two layers.
src_denoising/
: Contains the codebase from the N3Net repository.- Remaining Directories: Comprise the transformed EDM code, adapted by our team to integrate the N3 Block into the DhariwalUNet architecture.
- Origin: Adapted from EDM's DhariwalUNet.
- Changes: Network dimensions were altered to fit the available hardware resources.
- Origin: DhariwalUNet from EDM.
- Enhancement: Replaced attention mechanisms at the first two layers with N3 Blocks.
- Reference: N3Net GitHub Repository.
- Modified EDM Code by Shadi.