moiduy04 commited on
Commit
d7c7d33
1 Parent(s): c0780b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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 preload and save models in `model_folder`
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`"