Skip to content

Commit

Permalink
Update CONTRIBUTING.md
Browse files Browse the repository at this point in the history
  • Loading branch information
haifeng-jin committed Sep 12, 2018
1 parent ef256bc commit 2b5cbf4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion CONTRIBUTING.md
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ there are some other classes you can extend.
### ModelTrainer
[`autokeras.model_trainer.ModelTrainer`](https://github.com/jhfjhfj1/autokeras/blob/master/autokeras/model_trainer.py) is a class for training Pytorch models.
If needed a new metric or loss function other than the ones we have, you can add your own to [`loss_function.py`](https://github.com/jhfjhfj1/autokeras/blob/master/autokeras/loss_function.py) and [`metric.py`](https://github.com/jhfjhfj1/autokeras/blob/master/autokeras/metric.py).
You can follow its [documentation]() to use it.
You can follow its [documentation](https://github.com/jhfjhfj1/autokeras/blob/master/autokeras/model_trainer.py) and this [example](https://github.com/jhfjhfj1/autokeras/blob/master/examples/code_reuse_example.py) to use it.
Make sure your loss function, metric, Pytorch model, and Dataloader are compatible with each other.

## Main Contributor List
Expand Down

0 comments on commit 2b5cbf4

Please sign in to comment.