subcenter arcface学习笔记

目录

有注释和调用示例：

代码2：

torch.bmm函数示例

---

有注释和调用示例：

https://github.com/ditwoo/kaggle-lyft-motion-prediction-autonomous-vehicles/blob/ec8be5c55f61fcc8274607de582c576f58b0485d/src/batteries/layers/arcface.py

import math

import torch
import torch.nn as nn
import torch.nn.functional as F


class ArcFace(nn.Module):
    """Implementation of
    `ArcFace: Additive Angular Margin Loss for Deep Face Recognition`_.
    .. _ArcFace\: Additive Angular Margin Loss for Deep Face Recognition:
        https://arxiv.o