Skip to content

Commit 0ccbc27

Browse files
committed
Update documentateion for pretraining
1 parent 1826aa0 commit 0ccbc27

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,20 @@ ChEBai is a deep learning library that allows the combination of deep learning
66
## Pretraining
77

88
```
9-
python -m chebai fit --config=[path-to-your-config] --trainer.callbacks=configs/training/default_callbacks.yml
9+
fit --data.class_path=chebai.preprocessing.datasets.pubchem.SWJChem --model=configs/model/electra-for-pretraining.ElectraPre --model.train_metrics=configs/metrics/micro-macro-f1.yml --model.val_metrics=configs/metrics/micro-macro-f1.yml --model.test_metrics=configs/metrics/micro-macro-f1.yml --trainer=configs/training/default_trainer.yml --trainer.callbacks=configs/training/default_callbacks.yml
1010
```
1111

1212
## Structure-based ontology extension
1313

1414
```
15-
python -m chebai train --config=[path-to-your-electra_chebi100-config] --trainer.callbacks=configs/training/default_callbacks.yml --model.pretrained_checkpoint=[path-to-pretrained-model] --model.load_prefix=generator.
15+
python -m chebai fit --config=[path-to-your-electra_chebi100-config] --trainer.callbacks=configs/training/default_callbacks.yml --model.pretrained_checkpoint=[path-to-pretrained-model] --model.load_prefix=generator.
1616
```
1717

1818

1919
## Fine-tuning for Toxicity prediction
2020

2121
```
22-
python -m chebai train --config=[path-to-your-tox21-config] --trainer.callbacks=configs/training/default_callbacks.yml --model.pretrained_checkpoint=[path-to-pretrained-model] --model.load_prefix=generator.
22+
python -m chebai fit --config=[path-to-your-tox21-config] --trainer.callbacks=configs/training/default_callbacks.yml --model.pretrained_checkpoint=[path-to-pretrained-model] --model.load_prefix=generator.
2323
```
2424

2525
```

0 commit comments

Comments
 (0)