最新消息: USBMI致力于为网友们分享Windows、安卓、IOS等主流手机系统相关的资讯以及评测、同时提供相关教程、应用、软件下载等服务。

【pytorch】RuntimeError: Adam does not support sparse gradients, please consider SparseAdam instead

业界 admin 35浏览 0评论

报错:
RuntimeError: Adam does not support sparse gradients, please consider SparseAdam instead

原因:在训练时使用nn.Embedding(sizes[0], dim, sparse=True),这里需要将sparse=True改为sparse=False

报错:
RuntimeError: Adam does not support sparse gradients, please consider SparseAdam instead

原因:在训练时使用nn.Embedding(sizes[0], dim, sparse=True),这里需要将sparse=True改为sparse=False

发布评论

评论列表 (0)

  1. 暂无评论