Update README.md
Browse files
README.md
CHANGED
@@ -54,6 +54,7 @@ train_model(config)
|
|
54 |
If you wish to continue training/ fine-tune our model, you should
|
55 |
modify the `num_epochs` in your desired config file,
|
56 |
as well as read the following notes (`+` is the string concat funtion):
|
57 |
-
- The code will
|
58 |
- The code will preload the model with the name: "`model_basename` + `preload` + `.pt`"
|
|
|
59 |
- Every epoch, the code will save the model with the name: "`model_basename` + `_` + (current epoch) + `.pt`"
|
|
|
54 |
If you wish to continue training/ fine-tune our model, you should
|
55 |
modify the `num_epochs` in your desired config file,
|
56 |
as well as read the following notes (`+` is the string concat funtion):
|
57 |
+
- The code will save and preload models in `model_folder`
|
58 |
- The code will preload the model with the name: "`model_basename` + `preload` + `.pt`"
|
59 |
+
- The code will NOT preload a trained model if you set `preload` as `null`
|
60 |
- Every epoch, the code will save the model with the name: "`model_basename` + `_` + (current epoch) + `.pt`"
|