文件名称:pytorch_metric_learning 一个灵活且可扩展的度量学习库,用PyTorch编写-python
文件大小:12.91MB
文件格式:ZIP
更新时间:2024-07-08 12:11:57
机器学习
pytorch_metric_learning 一个灵活且可扩展的度量学习库,用PyTorch编写 PyTorch Metric Learning News 6 月 20 日:v0.9.87 带来了一些可能导致您现有代码中断的重大更改。 有关详细信息,请参阅发行说明。 文档 在此处查看文档 Google Colab 示例 请参阅笔记本的示例文件夹,其中显示了带有日志记录和模型保存的整个训练/测试工作流程。 该库的优点 易于使用 在您的训练循环中只需 2 行代码即可将度量学习添加到您的应用程序中。 使用单个函数调用挖掘对和三元组。 灵活性以其他图书馆不允许的方式混合和匹配损失、矿工和培训师。 安装 Pip pip install pytorch-metric-learning 要获得最新的开发版本: pip install pytorch-metric-learning==0.9.88 在 Windows 上安装: pip install torch===1.4.0 torchvision===0.5.0 - f https://download.pytorch.org/wh