quaterion.train.xbm.xbm_config 模块

class XbmConfig(weight: float | None = 1.0, buffer_size: int | None = 10000, start_iteration: int | None = 1000, device: XbmDevice | None = XbmDevice.AUTO)

基类: object

确定 XBM 设置。

此类应从 configure_xbm() 返回

buffer_size: int | None = 10000

内存缓冲区的大小，用于保存先前批次的 embeddings

device: XbmDevice | None ='auto'

缓冲区的位置

start_iteration: int | None = 1000

开始考虑缓冲区损失的迭代步骤

weight: float | None = 1.0

在将缓冲区损失添加到最终损失之前用于缩放的值

class XbmDevice(value)

基类: str, Enum

用于放置缓冲区的设备选择

AUTO = 'auto'

如果 GPU 可用，则在 GPU 中创建缓冲区。否则在 CPU 中创建。

CPU = 'cpu'

在 CPU 中创建缓冲区

CUDA = 'cuda'

在 GPU 中创建缓冲区