Update README.md
Browse files
README.md
CHANGED
|
@@ -25,6 +25,7 @@ datasets:
|
|
| 25 |
Nguyen Dinh Hieu [0009-0002-6683-8036], elt..
|
| 26 |
**Institution:** FPT University, Hanoi, Vietnam
|
| 27 |
📧 hieundhe180318@fpt.edu.vn
|
|
|
|
| 28 |
Full code at: https://github.com/nguyendinhhieu1309/EquiFashion.git
|
| 29 |
## 🧩 Overview
|
| 30 |
|
|
@@ -78,7 +79,7 @@ print(dataset)
|
|
| 78 |
| Setting | Value |
|
| 79 |
|----------|-------|
|
| 80 |
| Framework | PyTorch Lightning 2.2 |
|
| 81 |
-
| GPU | NVIDIA A100 (40 GB, CUDA 12.
|
| 82 |
| Optimizer | AdamW |
|
| 83 |
| Learning Rate | 2e-4 (G), 1e-4 (D) |
|
| 84 |
| Scheduler | Cosine Decay |
|
|
|
|
| 25 |
Nguyen Dinh Hieu [0009-0002-6683-8036], elt..
|
| 26 |
**Institution:** FPT University, Hanoi, Vietnam
|
| 27 |
📧 hieundhe180318@fpt.edu.vn
|
| 28 |
+
|
| 29 |
Full code at: https://github.com/nguyendinhhieu1309/EquiFashion.git
|
| 30 |
## 🧩 Overview
|
| 31 |
|
|
|
|
| 79 |
| Setting | Value |
|
| 80 |
|----------|-------|
|
| 81 |
| Framework | PyTorch Lightning 2.2 |
|
| 82 |
+
| GPU | NVIDIA A100 (40 GB, CUDA 12.8) |
|
| 83 |
| Optimizer | AdamW |
|
| 84 |
| Learning Rate | 2e-4 (G), 1e-4 (D) |
|
| 85 |
| Scheduler | Cosine Decay |
|