Skip to content

Commit c913099

Browse files
authored
update on instruction for transfer learning
1 parent d2a819b commit c913099

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

elemnet/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ ElemNet is a 17-layered fully connected network for the prediction of formation
2323

2424
* `sample`: A sample run folder that contains running configuration and the
2525
ElemNet trained using random split of training-data/oqmd_all-22Mar18.csv. The
26-
'sample_model' can be used for transfer learning.
26+
'sample_model' can be used for transfer learning to replicate the results in paper [2].
2727

2828
* `make_prediction.ipynb`: Jupyter notebook that illustrates how to make predictions using the trained ElemNet model from paper [1].
2929

@@ -33,7 +33,7 @@ You can simply run the code by passing a sample config file to the dl_regressors
3333

3434
`python dl_regressors.py --config_file sample/sample-run.config`
3535

36-
The config file defines the loss_type, training_data_path, test_data_path, label, input_type [elements_tl for ElemNet] and other runtime parameters. For transfer learning, you need to set 'model_path' [e.g. `sample/sample_model`]. The output log
36+
The config file defines the loss_type, training_data_path, test_data_path, label, input_type [elements_tl for ElemNet] and other runtime parameters. For transfer learning used in paper [2], you need to set 'model_path' to the model checkpoint trained on the larger dataset (OQMD in our case) [e.g. `"model_path":"sample/sample_model"`] in the config file. The output log
3737
from this sample run is provided in the `sample/sample.log` file.
3838

3939
* Extra files:

0 commit comments

Comments
 (0)