
PyTorch-Metric-Learning:轻松将深度度量学习应用于程序的最佳选择。模块化、灵活且易于扩展。用PyTorch...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
PyTorch-Metric-Learning是一个高效的深度度量学习库,基于PyTorch构建,提供模块化和高度灵活的框架,方便用户轻松集成到项目中并进行扩展。
新闻1月12日:v0.9.96大大提高了功能的灵活性。
查看12月10日:v0.9.95包括一个新的元组采矿器。
查看11月6日:v0.9.94中包含了一些小错误修复和改进。
PyTorch公制学习概述
该库包含了九个模块,每个模块都可以在您现有的代码库中独立使用或组合在一起以形成完整的训练/测试工作流程。损失函数如何运作,在训练循环中利用损失和矿工:
首先初始化一个简单的例子:
```python
from pytorch_metric_learning import losses
loss_func = losses.TripletMarginLoss()
```
为了计算训练循环中的损失,需要传递由模型生成的嵌入以及相应的标签。这些嵌入的大小应为(N, embedding_size),而标签的大小则应该是(N),其中N代表批次大小。
```python
# 训练循环示例
for i, (data, labels) in enumerate(dataloader):
# 执行模型计算,得到嵌入和预测值等操作...
```
在上述代码中,“dataloader”是数据加载器的一个实例。
全部评论 (0)
还没有任何评论哟~


