prelington commited on
Commit
c264cbb
·
verified ·
1 Parent(s): 55c08f6

Create config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +21 -0
config.yaml ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ name: "OrcaleSeek"
3
+ version: "1.0.0"
4
+ type: "text-classification"
5
+ architecture: "transformer"
6
+
7
+ training:
8
+ learning_rate: 2e-5
9
+ batch_size: 16
10
+ max_sequence_length: 512
11
+ num_epochs: 5
12
+
13
+ inference:
14
+ max_length: 128
15
+ temperature: 0.7
16
+ top_p: 0.9
17
+ do_sample: true
18
+
19
+ deployment:
20
+ supported_frameworks: ["pytorch", "onnx"]
21
+ min_memory_gb: 4