Hugging Face's logo

benetraco
/
brain_ddpm_64

Unconditional Image Generation
Diffusers
Safetensors
PyTorch
DDPMPipeline
diffusion-models-class
medical-imaging
brain-mri
multiple-sclerosis
Model card Files
xet
 Community
benetraco commited on
Commit
cd0724e
·
verified ·
1 Parent(s): 5ffe17c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ It is trained from scratch on a dataset of brain MRI slices, specifically design
29
  - **Gradient Accumulation:** 1 steps
30
  - **Mixed Precision:** No
31
  - **Hardware:** Trained on **one NVIDIA GeForce GTX 1080 Ti GPU of 12GB**
32
- - **Memory Consumption:** Around **11 GB** during training
33
 
34
  ## U-Net Architecture
35
  - **Down Blocks:** [DownBlock2D, DownBlock2D, AttnDownBlock2D, DownBlock2D]
 
29
  - **Gradient Accumulation:** 1 steps
30
  - **Mixed Precision:** No
31
  - **Hardware:** Trained on **one NVIDIA GeForce GTX 1080 Ti GPU of 12GB**
32
+ - **Memory Consumption:** Around **7 GB** during training
33
 
34
  ## U-Net Architecture
35
  - **Down Blocks:** [DownBlock2D, DownBlock2D, AttnDownBlock2D, DownBlock2D]