Upload README.md with huggingface_hub
Browse files
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 **
|
| 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]
|