| |
| pretrained_model_name: '3DNN' |
| load_weights: true |
| custom_model: false |
|
|
| |
| window_size: 300 |
| stride: 300 |
| normalisation: 'minmax' |
| sampling_frequency: 30 |
| sample_data_path: .sample_data |
|
|
|
|
| |
| tensordataset: true |
| batch_size: 4 |
| num_workers: 2 |
| shuffle: true |
| device: cuda:0 |
|
|
| |
| model_cache_dir: '.cache_dir' |
| repo_id: "viks66/TVB" |
|
|
| |
| plot_folder: .plots |
| use_logger: true |
| metrics_verbose: 1 |
| progressbar: false |
|
|
| |
| finetune: true |
| lr: 0.001 |
| epochs: 1000 |
| save_chk_folder: test_model |
| chk_name: 'chk.pth' |
| random_segment_training: true |
| min_segment_length: 3 |
| save_by_validation_chk: true |
| overwrite_model: false |
|
|
