Update README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ Then use `load_model_tokenizer` to load a pretrained model and its tokenizers
|
|
19 |
from config import load_config
|
20 |
from load_model import load_model_tokenizer
|
21 |
|
22 |
-
config = load_config(file_name='
|
23 |
model, src_tokenizer, tgt_tokenizer = load_model_tokenizer(config)
|
24 |
```
|
25 |
|
@@ -30,7 +30,7 @@ from translate import translate
|
|
30 |
from config import load_config
|
31 |
from load_model import load_model_tokenizer
|
32 |
|
33 |
-
config = load_config(file_name='
|
34 |
model, src_tokenizer, tgt_tokenizer = load_model_tokenizer(config)
|
35 |
|
36 |
text = " "
|
@@ -47,7 +47,7 @@ Use the `train_model` function in `train.py` to train your model.
|
|
47 |
from train import train_model
|
48 |
from load_model import load_model_tokenizer
|
49 |
|
50 |
-
config = load_config(file_name='config_final.yaml')
|
51 |
train_model(config)
|
52 |
```
|
53 |
|
|
|
19 |
from config import load_config
|
20 |
from load_model import load_model_tokenizer
|
21 |
|
22 |
+
config = load_config(file_name='config/config_final.yaml')
|
23 |
model, src_tokenizer, tgt_tokenizer = load_model_tokenizer(config)
|
24 |
```
|
25 |
|
|
|
30 |
from config import load_config
|
31 |
from load_model import load_model_tokenizer
|
32 |
|
33 |
+
config = load_config(file_name='config/config_final.yaml')
|
34 |
model, src_tokenizer, tgt_tokenizer = load_model_tokenizer(config)
|
35 |
|
36 |
text = " "
|
|
|
47 |
from train import train_model
|
48 |
from load_model import load_model_tokenizer
|
49 |
|
50 |
+
config = load_config(file_name='config/config_final.yaml')
|
51 |
train_model(config)
|
52 |
```
|
53 |
|