dmlc--dgl
Mufei Li
cd9fb7baa0
* [Model zoo] Model zoo (#765) * tox21 * fix ci * fix ci * fix urls to url * add doc * remove binary * model zoo * test * markdown * fix typo * fix typo * fix typo * raise error * fix lint * remove unnecessary * fix doc * fix * fix * fix * fix * fix * fix * Update * CI * Fix * Fix * Fix * Fix * Fix * CI
16 行
571 B
Markdown
16 行
571 B
Markdown
Model Zoo API
|
|
==================
|
|
|
|
We provide two major APIs for the model zoo. For the time being, only PyTorch is supported.
|
|
- `model_zoo.chem.[Model_Name]` to load the model skeleton
|
|
- `model_zoo.chem.load_pretrained([Pretrained_Model_Name])` to load the model with pretrained weights
|
|
|
|
Models would be placed in `python/dgl/model_zoo/chem`.
|
|
|
|
Each Model should contain the following elements:
|
|
- Papers related to the model
|
|
- Model's input and output
|
|
- Dataset compatible with the model
|
|
- Documentation for all the customizable configs
|
|
- Credits (Contributor infomation)
|