Mustafa Mohamed commited on
Commit
a12689d
1 Parent(s): 71fca58

Update README.md

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