Skip to content

rotematari/DDPM_with_N3Block

Repository files navigation

DDPM_with_N3Block: Integrating N3 Blocks with DhariwalUNet for Enhanced Performance

Overview

This repository contains the implementation of two distinct neural network architectures aimed at improving existing models:

  1. A modified version of the baseline model from the Exemplar Diffusion Machine (EDM), specifically the DhariwalUNet. Modifications were made to accommodate hardware constraints.
  2. An advanced version of the DhariwalUNet that incorporates N3 Blocks from N3Net, replacing the attention mechanisms at the initial two layers.

Directory Structure

  • 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.

Modifications

Baseline Model

  • Origin: Adapted from EDM's DhariwalUNet.
  • Changes: Network dimensions were altered to fit the available hardware resources.

Enhanced Model

  • Origin: DhariwalUNet from EDM.
  • Enhancement: Replaced attention mechanisms at the first two layers with N3 Blocks.
  • Reference: N3Net GitHub Repository.

Contributors

  • Modified EDM Code by Shadi.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages