Upload config.yaml with huggingface_hub
Browse files- config.yaml +3 -3
config.yaml
CHANGED
|
@@ -30,7 +30,7 @@ discretization: null
|
|
| 30 |
enable_amp: false
|
| 31 |
enable_scaler: false
|
| 32 |
entity_kg_num_interval: '[0,inf)'
|
| 33 |
-
epochs:
|
| 34 |
eval_args:
|
| 35 |
group_by: user
|
| 36 |
mode:
|
|
@@ -47,7 +47,7 @@ filter_inter_by_user_or_item: true
|
|
| 47 |
gpu_id: '0'
|
| 48 |
hidden_act: gelu
|
| 49 |
hidden_dropout_prob: 0.5
|
| 50 |
-
hidden_size:
|
| 51 |
initializer_range: 0.02
|
| 52 |
inner_size: 256
|
| 53 |
item_inter_num_interval: '[5,inf)'
|
|
@@ -87,7 +87,7 @@ save_dataset: true
|
|
| 87 |
seed: 2020
|
| 88 |
seq_len: null
|
| 89 |
seq_separator: ' '
|
| 90 |
-
show_progress:
|
| 91 |
shuffle: true
|
| 92 |
single_spec: true
|
| 93 |
state: INFO
|
|
|
|
| 30 |
enable_amp: false
|
| 31 |
enable_scaler: false
|
| 32 |
entity_kg_num_interval: '[0,inf)'
|
| 33 |
+
epochs: 100
|
| 34 |
eval_args:
|
| 35 |
group_by: user
|
| 36 |
mode:
|
|
|
|
| 47 |
gpu_id: '0'
|
| 48 |
hidden_act: gelu
|
| 49 |
hidden_dropout_prob: 0.5
|
| 50 |
+
hidden_size: 16
|
| 51 |
initializer_range: 0.02
|
| 52 |
inner_size: 256
|
| 53 |
item_inter_num_interval: '[5,inf)'
|
|
|
|
| 87 |
seed: 2020
|
| 88 |
seq_len: null
|
| 89 |
seq_separator: ' '
|
| 90 |
+
show_progress: false
|
| 91 |
shuffle: true
|
| 92 |
single_spec: true
|
| 93 |
state: INFO
|