You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

t_backend.rst 877 B

4 years ago
4 years ago
4 years ago
1234567891011121314151617181920212223242526272829303132333435
  1. .. _backend:
  2. Backend Support
  3. ===============
  4. Currently, AutoGL support both pytorch geometric backend and deep graph library backend to
  5. enable users from both end benifiting the automation of graph learning.
  6. To specify one specific backend, you can declare the backend using environment variables
  7. ``AUTOGL_BACKEND``. For example:
  8. .. code-block:: python
  9. AUTOGL_BACKEND=pyg python xxx.py
  10. or
  11. .. code-block:: python
  12. import os
  13. os.environ["AUTOGL_BACKEND"] = "pyg"
  14. import autogl
  15. ...
  16. If no backend is specified, AutoGL will use the backend in your environment. If you have both
  17. Deep Graph Library and PyTorch Geometric installed, the default backend will be Deep Graph Library.
  18. You can also get current backend in the code by:
  19. .. code-block :: python
  20. from autogl.backend import DependentBackend
  21. print(DependentBackend.get_backend_name())