src is not broadcastable to dst, but they have the same number of elements
解决方法:
own_state[name].copy_(param)
改之后:
own_state[name].copy_(param.view_as(own_state[name]))
src is not broadcastable to dst, but they have the same number of elements
解决方法:
own_state[name].copy_(param)
改之后:
own_state[name].copy_(param.view_as(own_state[name]))