Weinan Gan

icon company Huawei Noah's Ark Lab icon location 中国深圳 icon location I am now a researcher at Noah's Ark Lab, Huawei. Recently focus on LLM Alignment & LLM Enhanced Information Retrieval

Results 2 issues of Weinan Gan

would u like to explain the batch beam search for transformer in details?

关于子类化构建模型build的问题

笔者对于继承model的自定义模型，采用model.build的方法，我有个问题，针对多输入问题，shape参数该如何写呢，目前没有看到相关方案。另外2.0对于继承model类的自定义模型若采用fit方法训练则可以保存，否则需要设置输入的形状，对于build方法，笔者有实验过可以正常保存模型嘛？谢谢