最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

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

运维笔记admin27浏览0评论

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

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

发布评论

评论列表(0)

  1. 暂无评论