Commit ·
7942c5b
1 Parent(s): 6a8192e
fix: model paths
Browse files- meg-gpt/config.yml +2 -2
meg-gpt/config.yml CHANGED
| @@ -26,7 +26,7 @@ model_config: | |
| 26 | n_channels: 52 |
| 27 | name: meg_gpt |
| 28 | sequence_length: 80 |
| 29 | - tokenizer_path: |
| 30 | training_config: |
| 31 | batch_size: 8 |
| 32 | n_epochs: 1 |
| @@ -34,4 +34,4 @@ training_config: | |
| 34 | learning_rate: 1.0e-05 |
| 35 | multi_gpu: True |
| 36 | checkpoint: |
| 37 | - save_dir: |
| | |
| 26 | n_channels: 52 |
| 27 | name: meg_gpt |
| 28 | sequence_length: 80 |
| 29 | + tokenizer_path: ./tokenizer |
| 30 | training_config: |
| 31 | batch_size: 8 |
| 32 | n_epochs: 1 |
| | |
| 34 | learning_rate: 1.0e-05 |
| 35 | multi_gpu: True |
| 36 | checkpoint: |
| 37 | + save_dir: ./ |