config params stage 01 update
Browse files- configs/config.yaml +1 -0
- params.yaml +10 -0
configs/config.yaml
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
|
| 3 |
artifacts:
|
| 4 |
ARTIFACTS_DIR: artifacts
|
|
|
|
| 5 |
TRAINED_MODEL_DIR: model
|
| 6 |
BASE_MODEL_DIR: base_model
|
| 7 |
BASE_MODEL_NAME: base_model.h5
|
|
|
|
| 2 |
|
| 3 |
artifacts:
|
| 4 |
ARTIFACTS_DIR: artifacts
|
| 5 |
+
cache_dir: artifacts/cache_dir
|
| 6 |
TRAINED_MODEL_DIR: model
|
| 7 |
BASE_MODEL_DIR: base_model
|
| 8 |
BASE_MODEL_NAME: base_model.h5
|
params.yaml
CHANGED
|
@@ -1 +1,11 @@
|
|
| 1 |
# This contains params to be used by the stages to train or predict
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# This contains params to be used by the stages to train or predict
|
| 2 |
+
train:
|
| 3 |
+
squad_v2: False
|
| 4 |
+
model_checkpoint: "distilbert-base-uncased"
|
| 5 |
+
batch_size: 16
|
| 6 |
+
max_length: 384 # The maximum length of a feature (question and context)
|
| 7 |
+
doc_stride: 128
|
| 8 |
+
learning_rate: 2e-5
|
| 9 |
+
num_train_epochs: 1
|
| 10 |
+
weight_decay: 0.01
|
| 11 |
+
dataset_name: 'squad'
|