wjybill

加载GEM pretrain model提示模型shape和权重shape不一致

1

你好，我在加载geognn模型预训练权重（位置在PaddleHelix/apps/pretrained_coumpound/ChemRL/GEM/pretrain_models/class.pdparams）的时候显示有两个layes的权重大小匹配不上，分别是 init_atom_embedding.embed_list.3 和 init_atom_embedding.embed_list.6。如下图所示  另外还有一个问题，如果想手动修改geognn模型中的每层的节点数量应该如何操作呢？