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

Nipponsha

Artificial intelligence AI: Keras PyTorch MXNet TensorFlow PaddlePaddle deep learning combat (updated from time to time)


 

        RuntimeError: Adam does not support sparse gradients, please consider SparseAdam instead
        分析:nn.Embedding(vocab_size, embed_dim, sparse=True)中不能等于True,必须为False
        解决:nn.Embedding(vocab_size, embed_dim, sparse=False)

 

Published 453 original articles · praised 155 · 230,000 views +

Guess you like

Origin blog.csdn.net/zimiao552147572/article/details/105500329