multi-label分类，loss一直增大

原創 guotong1988 2020-02-27 00:07

label为[batch_size, num_class]
logits为[batch_size, num_class]

每个label为比如[0,0,1,0,0,0,1,0,1,0]，就是10类有3类正确

不能用tf.nn.softmax_cross_entropy_with_logits

Pytorch使用torch.nn.BCEloss
Tensorflow使用tf.losses.sigmoid_cross_entropy

發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章

TensorFlow Hub-人人可享的易用模型

構建機器學習模型不但需要豐富的行業經驗，還需要存取海量數據和計算機資源，這對大多數開發者來說都是道不易翻越的屏障。爲解決這一問題，開發者可以使用預訓練過的模型，而且這種解決方案要好得多。TensorFlow Hub 就是爲解決此難題而設的庫

王莹 2021-12-02 13:08:54

TensorFlow Recommenders 一个专为搭建推荐系统设计的 TensorFlow 库

從電商到短視頻，推薦系統唄被廣泛應用於各個場景，但實際上要打造好一個有效的推薦系統並不容易。TensorFlow Recommenders 就是這樣一個爲了打造高效可擴展的推薦系統而專門製作的庫。在本次專題演講中，我們將通過演示如何使用 T

王莹 2021-12-01 10:13:50

打造负责任的 AI 工具包

隨着全世界數十億人繼續使用內核含 AI 技術的產品或服務，現在比以往更迫切需要負責任地使用 AI。打造有益於社會、包容且對社區負責的產品一直是我們最重視的事項。在本次專題演講中，我們將簡要介紹 Responsible AI Toolkit

王莹 2021-11-30 09:48:49

基于图像的机器学习技术将数十亿的电子商务产品分为数千个类别

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

Celian Gossec 2021-11-29 16:28:50

2021 Google开发者大会精彩回顾

{"type":"doc","content":[{"type":"heading","attrs":{"align":null,"level":3}},{"type":"paragraph","attrs":{"indent":0,"nu

王莹 2021-11-22 10:08:54

基于 TensorFlow.js 的下一代 Web 应用

TensorFlow.js介紹，功能演示，新動向以及社區活動。 直達 Google 開發者在線課程，收穫 TensorFlow 技術乾貨 https://developers.google.cn/learn/pathways?utm_sou

谷歌 2021-11-18 09:28:56

利用 TensorFlow Lite Model Maker 打造简单易用的端上模型

近年來，終端上的機器學習被廣泛應用在現實的移動應用當中。如何創建、優化、部署端上機器學習模型是TensorFlow Lite的主要關注點之一。我們開發了許多端到端的工具鏈來解決移動開發人員和機器學習初學者最爲頭疼的問題。今天，我們將介紹Te

谷歌 2021-11-18 09:28:56

2021 Google 开发者大会一览，同步Android、TensorFlow、Web开发等最新动态

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

罗燕珊 2021-11-16 19:33:53

腾讯发布超大预训练系统派大星，聚焦解决BERT等超大模型训练时的“GPU内存墙”问题

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

InfoQ编辑部 2021-11-02 13:38:53

谷歌推出面向开发者的全新开源平台Dev Library

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

Tina 2021-10-19 16:33:48

突破 PyTorch、TensorFlow 并行瓶颈的开源训练加速框架到底是啥？

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

赵钰莹 2021-10-12 09:38:58

TensorFlow出现任意代码执行漏洞，团队宣布撤销对YAML的支持

{"type":"doc","content":[{"type":"blockquote","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null

Ben Dickson 2021-09-02 08:03:56

阿里巴巴AI系统建设的思考

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

林伟 2021-07-28 15:58:54

快手八卦！突破TensorFlow、PyTorch并行瓶颈的开源分布式训练框架来了！

{"type":"doc","content":[{"type":"blockquote","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null

快手AI团队 2021-07-16 14:03:55

基于 TensorFlow.js 和 MoveNet 的下一代姿态检测

{"type":"doc","content":[{"type":"paragraph","attrs":{"indent":0,"number":0,"align":null,"origin":null},"content":[{"typ

TensorFlow官博 2021-06-08 09:23:55