Mustafa Mohamed
commited on
Commit
•
a12689d
1
Parent(s):
71fca58
Update README.md
Browse files
README.md
CHANGED
@@ -3,4 +3,38 @@ license: mit
|
|
3 |
datasets:
|
4 |
- mstafam/Kilter-Board-Dataset
|
5 |
pipeline_tag: text2text-generation
|
6 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
datasets:
|
4 |
- mstafam/Kilter-Board-Dataset
|
5 |
pipeline_tag: text2text-generation
|
6 |
+
---
|
7 |
+
|
8 |
+
# GenClimb: AI-Generated Climbing Routes for Interactive Training Boards
|
9 |
+
|
10 |
+
GenClimb is a generative AI model designed to create climbing routes for Standardized Interactive Climbing Training Boards (SICTBs). It a seq2seq transformer architecture, GenClimb generates climbs based on board layouts and climb difficulties.
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Architecture
|
15 |
+
|
16 |
+
- **Dimension**: 512
|
17 |
+
- **Attention Heads**: 4
|
18 |
+
- **Layers**: 5
|
19 |
+
- **Feed-Forward Dimension**: 1024
|
20 |
+
- **Dropout Rate**: 0.15
|
21 |
+
- **Activation Function**: GELU
|
22 |
+
- **Layer Normalization Epsilon**: 1e-5
|
23 |
+
|
24 |
+
### Training Configuration
|
25 |
+
|
26 |
+
- **Device**: CUDA (1x NVIDIA GeForce RTX 3070)
|
27 |
+
- **Learning Rate**: 1e-4
|
28 |
+
- **Epochs**: 8
|
29 |
+
- **Weight Decay**: 0.0125
|
30 |
+
- **Batch Size**: 32
|
31 |
+
- **Train/Test Split**: 90/10
|
32 |
+
|
33 |
+
### Performance Metrics
|
34 |
+
|
35 |
+
- **Training Time**: 12 hours and 6 minutes
|
36 |
+
- **Final Loss**: 2.114803
|
37 |
+
|
38 |
+
## Dataset
|
39 |
+
|
40 |
+
The model is trained on the [Kilter-Board-Dataset](https://huggingface.co/datasets/mstafam/Kilter-Board-Dataset), a comprehensive collection of climbing routes curated with the help of the [BoardLib utility](https://github.com/lemeryfertitta/BoardLib) by lemeryfertitta.
|