quaterion.loss.fast_ap_loss 模块¶
- 类 FastAPLoss(num_bins: int | None = 10)[源代码]¶
基类：
GroupLossFastAP 损失
改编自 https://github.com/kunhe/FastAP-metric-learning。
- 更多信息
https://cs-people.bu.edu/fcakir/papers/fastap_cvpr2019.pdf。"Deep Metric Learning to Rank" Fatih Cakir(*), Kun He(*), Xide Xia, Brian Kulis, and Stan Sclaroff IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2019
- 参数：
num_bins – 用于计算平均精度的软直方图 bin 数量。论文建议使用 10。
- forward(embeddings: Tensor, groups: Tensor) Tensor[源代码]¶
计算损失值。
- 参数：
embeddings – 形状：(batch_size, vector_length) - 嵌入批次。
groups – 形状：(batch_size,) - 与 embeddings 关联的标签批次。
- 返回：
Tensor – 标量损失值。
- get_config_dict() Dict[str, Any][源代码]¶
用于保存和加载的配置。
配置对象必须是 JSON 可序列化的。
- 返回：
Dict[str, Any] – 可 JSON 序列化的参数字典
- training: bool¶
