diff --git a/mkdocs.yml b/mkdocs.yml index 157f7d923c..98fa5559c1 100644 --- a/mkdocs.yml +++ b/mkdocs.yml @@ -2,6 +2,7 @@ site_name: PyTorch lightning Documentation theme: name: 'material' docs_dir: docs +repo_name: 'williamFalcon/pytorch-lightning' repo_url: https://github.com/williamFalcon/pytorch-lightning site_dir: 'site' site_description: 'Documentation for PyTorch LightningModule, the researcher version of keras.'