Skip to content

Commit 2fccd85

Browse files
committed
Add link to Pytorch
This gives everyone a heads up that they will need Pytorch to run this example.
1 parent 7f1e99b commit 2fccd85

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

examples/darts/README.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,14 @@ components as the model is training.
6565
As the optimization runs, the model will print the resulting loss with respect to a given *Genotype*.
6666
The final model will be the *Genotype* with corresponding to the lowest loss.
6767

68+
Setup
69+
-----
70+
71+
Darts makes use of Pytorch. You can find binaries for both Pytorch and Torchvision (used in the advanced
72+
example) at the `pytorch website`_.
73+
6874
.. References
6975
.. ----------
7076
.. _paper: https://openreview.net/forum?id=S1eYHoC5FX
7177
.. _darts.modules.operations.conv.py: ../../../common/darts/modules/operations/conv.py
78+
.. _pytorch website: https://pytorch.org/

0 commit comments

Comments
 (0)