什么是gpytorch.settings.max_preconditioner_size?以及gpytorch.beta_features.checkpoint_kernel(checkpoint_size)是什么?

2024-06-16 08:28:14 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在学习教程Simple_MultiGPU_GP_Regression,我注意到在培训期间建立了两个选项:

    with gpytorch.beta_features.checkpoint_kernel(checkpoint_size), \
         gpytorch.settings.max_preconditioner_size(preconditioner_size):

变量checkpoint_kernelpreconditioner_size是什么或指什么

我已经检查了文档preconditioner_size,但我不太清楚它指的是什么

我直觉地认为checkpoint_size指的是与每个GPU关联的训练点数相关的东西。但这只是一种直觉

谢谢你的帮助

谢谢


Tags: size选项with教程simplekernelbetafeatures