Backend Support

Currently, AutoGL support both pytorch geometric backend and deep graph library backend to enable users from both end benifiting the automation of graph learning.

To specify one specific backend, you can declare the backend using environment variables AUTOGL_BACKEND. For example:



import os
os.environ["AUTOGL_BACKEND"] = "pyg"
import autogl


If no backend is specified, AutoGL will use the backend in your environment. If you have both Deep Graph Library and PyTorch Geometric installed, the default backend will be Deep Graph Library.

You can also get current backend in the code by:

from autogl.backend import DependentBackend