def _convert_weight(weight, nw_size):
w_size = weight.shape
assert len(w_size) == len(nw_size)
w_ratio = [_nw / _w for _nw, _w in zip(nw_size, w_size)]
new_weight = scipy.ndimage.zoom(weight, w_ratio)
return new_weight
评论列表
文章目录
