Upload model checkpoint and configuration - config
Browse files- configuration.yaml +20 -3
configuration.yaml
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
jobname:
|
| 2 |
seed: 42
|
| 3 |
checkpoint: null
|
| 4 |
limit_batches: null
|
|
@@ -8,7 +8,7 @@ path:
|
|
| 8 |
hidden_states: /path/to/hidden-states
|
| 9 |
shs_data: /path/to/shs_data.csv
|
| 10 |
shs_splits: /path/to/shs/splits
|
| 11 |
-
mmeta: /
|
| 12 |
save_data_path: /path/to/data
|
| 13 |
data: /path/to/data
|
| 14 |
working_dir: /path/to/project
|
|
@@ -31,8 +31,20 @@ data:
|
|
| 31 |
max_length: 4000
|
| 32 |
use_random_chunks: true
|
| 33 |
chunk_size: 1500
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
model:
|
| 35 |
-
name:
|
| 36 |
input_channels: 384
|
| 37 |
sbert_hidden_layers: 2
|
| 38 |
conv_blocks: 0
|
|
@@ -66,3 +78,8 @@ early_stopping:
|
|
| 66 |
metric: m_MAP
|
| 67 |
mode: max
|
| 68 |
min_delta: 0.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
jobname: wealy_model
|
| 2 |
seed: 42
|
| 3 |
checkpoint: null
|
| 4 |
limit_batches: null
|
|
|
|
| 8 |
hidden_states: /path/to/hidden-states
|
| 9 |
shs_data: /path/to/shs_data.csv
|
| 10 |
shs_splits: /path/to/shs/splits
|
| 11 |
+
mmeta: /path/to/mmeta
|
| 12 |
save_data_path: /path/to/data
|
| 13 |
data: /path/to/data
|
| 14 |
working_dir: /path/to/project
|
|
|
|
| 31 |
max_length: 4000
|
| 32 |
use_random_chunks: true
|
| 33 |
chunk_size: 1500
|
| 34 |
+
path:
|
| 35 |
+
cache: /path/to/cache
|
| 36 |
+
logs: /path/to/logs
|
| 37 |
+
working_dir: /path/to/project
|
| 38 |
+
data: /path/to/data
|
| 39 |
+
save_data_path: /path/to/data
|
| 40 |
+
hidden_states: /path/to/hidden-states
|
| 41 |
+
meta: /path/to/metadata.pt
|
| 42 |
+
shs_data: /path/to/shs_data.csv
|
| 43 |
+
shs_splits: /path/to/shs/splits
|
| 44 |
+
lyric_covers_data: /path/to/lyric-covers
|
| 45 |
+
discogs_vi_data: /path/to/discogs-vi
|
| 46 |
model:
|
| 47 |
+
name: wealy
|
| 48 |
input_channels: 384
|
| 49 |
sbert_hidden_layers: 2
|
| 50 |
conv_blocks: 0
|
|
|
|
| 78 |
metric: m_MAP
|
| 79 |
mode: max
|
| 80 |
min_delta: 0.0
|
| 81 |
+
pytorch:
|
| 82 |
+
cudnn_benchmark: false
|
| 83 |
+
cudnn_deterministic: true
|
| 84 |
+
float32_matmul_precision: medium
|
| 85 |
+
detect_anomaly: false
|