特牛网址导航

解决mxnet报错:Embedding layer doesn't support calculate data gradient_error: layer does not support extent calculation-CSDN博客

网友收藏
文章浏览阅读538次。word_embed = mx.sym.Embedding(data=seq, input_dim=vocab_size, output_dim=num_embed, name='seq_embed')    改为:word_embed = mx.sym.Embedding(data=mx.sym.BlockGrad(seq), input_dim=vocab_size, output_d..._error: layer does not support extent calculation