Upload default.yaml
Browse files- default.yaml +40 -0
default.yaml
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#model
|
| 2 |
+
pretrained_model_name: '3DNN'
|
| 3 |
+
load_weights: true
|
| 4 |
+
custom_model: false
|
| 5 |
+
|
| 6 |
+
#data
|
| 7 |
+
window_size: 300
|
| 8 |
+
stride: 300
|
| 9 |
+
normalisation: 'minmax'
|
| 10 |
+
sampling_frequency: 30
|
| 11 |
+
sample_data_path: .sample_data
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
#dataloader
|
| 15 |
+
tensordataset: true
|
| 16 |
+
batch_size: 4
|
| 17 |
+
num_workers: 2
|
| 18 |
+
shuffle: true
|
| 19 |
+
device: cuda:0
|
| 20 |
+
|
| 21 |
+
#huggingface
|
| 22 |
+
model_cache_dir: '.cache_dir'
|
| 23 |
+
repo_id: "viks66/TVB"
|
| 24 |
+
|
| 25 |
+
#logs
|
| 26 |
+
plot_folder: .plots
|
| 27 |
+
use_logger: true
|
| 28 |
+
metrics_verbose: 1
|
| 29 |
+
progressbar: false
|
| 30 |
+
|
| 31 |
+
#training
|
| 32 |
+
finetune: true
|
| 33 |
+
lr: 0.001
|
| 34 |
+
epochs: 1000
|
| 35 |
+
save_chk_folder: test_model
|
| 36 |
+
chk_name: 'chk.pth'
|
| 37 |
+
random_segment_training: true
|
| 38 |
+
min_segment_length: 3
|
| 39 |
+
save_by_validation_chk: true
|
| 40 |
+
overwrite_model: false
|