Update README.md

README.md

@@ -45,7 +45,7 @@ print(translation)
 Use the `train_model` function in `train.py` to train your model.
 ```
 from train import train_model
-from load_model import load_model_tokenizer
+from config import load_config
 
 config = load_config(file_name='config/config_final.yaml')
 train_model(config)
@@ -57,4 +57,5 @@ as well as read the following notes (`+` is the string concat funtion):
 - The code will save and preload models in `model_folder`
 - The code will preload the model with the name: "`model_basename` + `preload` + `.pt`"
 - The code will NOT preload a trained model if you set `preload` as `null`
-- Every epoch, the code will save the model with the name: "`model_basename` + `_` + (current epoch) + `.pt`"
+- Every epoch, the code will save the model with the name: "`model_basename` + `_` + (current epoch) + `.pt`"
+- `train_model` will automatically continue training the `preload`ed model.