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())