NguyenDinhHieu commited on
Commit
8f54b18
·
verified ·
1 Parent(s): cb17de6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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.2) |
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 |