bam8pm commited on
Commit
43ad797
·
verified ·
1 Parent(s): af5159c

Formatted from .yaml to .json

Browse files
Files changed (2) hide show
  1. config.json +21 -0
  2. config.yaml +0 -19
config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "seed": 42,
3
+ "train": {
4
+ "batch_size": 32,
5
+ "epochs": 20,
6
+ "hidden_size": 64,
7
+ "dropout_prob": 0.13,
8
+ "learning_rate": 0.0075,
9
+ "l2_reg_lambda": 0.001,
10
+ "optimizer": "Adam",
11
+ "loss": "CrossEntropyLoss"
12
+ },
13
+ "tune": {
14
+ "n_trials": 50,
15
+ "hidden_size_choices": [32, 64, 128, 256, 512],
16
+ "dropout_prob_range": [0.1, 0.5],
17
+ "learning_rate_range": [1e-5, 1e-2]
18
+ }
19
+ }
20
+
21
+
config.yaml DELETED
@@ -1,19 +0,0 @@
1
- seed: 42
2
-
3
- train:
4
- batch_size: 32
5
- epochs: 20
6
- hidden_size: 64
7
- dropout_prob: 0.13
8
- learning_rate: 0.0075
9
- l2_reg_lambda: 0.001
10
- optimizer: Adam
11
- loss: CrossEntropyLoss
12
-
13
- tune:
14
- n_trials: 50
15
- hidden_size_choices: [32,64,128,256,512]
16
- dropout_prob_range: [0.1, 0.5]
17
- learning_rate_range: [1e-5, 1e-2]
18
-
19
-