Backend Support

目前,AutoGL支持使用PyTorch-Geometric或Deep Graph Library作为后端,以便熟悉两者之一的用户均可受益于自动图学习。

为指定特定的后端,用户可以使用环境变量``AUTOGL_BACKEND``进行声明,例如:

AUTOGL_BACKEND=pyg python xxx.py

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

...

如果环境变量``AUTOGL_BACKEND``未声明,AutoGL会根据用户的Python运行环境中所安装的图学习库自动选择。 如果PyTorch-Geometric和Deep Graph Library均已安装,则Deep Graph Library将被作为默认的后端。

可以以编程方式获得当前使用的后端:

from autogl.backend import DependentBackend
print(DependentBackend.get_backend_name())