embedding propagation:嵌入传播的代码库 源码

嵌入传播 适用于少量分类的平滑歧管 (ECCV2020) 嵌入传播可用于规范化中间特征,从而提高了泛化性能。 用法 将嵌入传播层添加到您的网络。 pip install git+https://github.com/ElementAI/embedding-propagation import torch from embedding_propagation import EmbeddingPropagation ep = EmbeddingPropagation () features = torch . randn ( 32 , 32 ) embeddings = ep ( features ) 实验 从生成结果。 安装要求 pip install -r requirements.txt 该命令将安装,该有助于管理实验。 下载数据集 () 如果你有pkl miniimagenet的版本