全部回复
LV.6
#2
三叶虫
回复
您好,因已超过15天,您侧未更新回复信息,此帖超时关闭,后续遇到问题,请您发起新帖,版主会及时分析处理。
2023-05-24 15:07:38
0
|
回复
LV.3
#1
HelloAI
回复
当
state得device为MLU时,不能直接赋值,如果要自己组合一个
argmax,可以将这个算子放在网络中,当model和input都整体都在MLU上是可以计算的,但不要有常量赋值操作
2023-05-05 17:05:41
5
|
回复
请
登录
后评论