我正在使用BucketingModule来训练多个小型模型/机器人。这里，桶密钥是bot_id。但是，每个机器人都有单独的一组目标标签/类（因此，每个机器人的softmax层大小不同）。
有没有办法在mxnet中训练这样的模型，我想在所有的机器人中分享所有层的权重，但是一个（softmax）？
我如何使用sym_gen方法初始化这样的模型？如果在sym_gen方法中，对于Softmax层我指定num_hidden=size_dict[bot]，即，
pred = mx.sym.FullyConnected(data=pred, num_hidden=len(size_dict[bot]), name='pred')
pred = mx.sym.SoftmaxOutput(data=pred, label=label, name='softmax')
我收到错误：
推断的形状与shared_exec.arg_array的形状不匹配
这是有意义的，因为每个机器人具有不同数量的目标类。
此问题已在此处发布并解决：https://github.com/apache/incubator-mxnet/issues/9042
您可以使sym_gen（default_bucket_key）返回包含所有这些不同形状的FC层的“主网络”，并且sym_gen（other_keys）返回具有一个特定FC的主网络的子集。请注意，对于主网络，您可能需要使用mx.sym.Group将所有输出组合在一起，因此只返回一个符号。