Function copy_from_param

SwissArmyTransformer/examples/cait/transform_param.py:37–39

(src, dst)

        assert (dst_dic[k].data == src_dic[k].data).all()

def copy_from_param(src, dst):
    assert src.data.shape == dst.data.shape
    dst.data = src.data

def copy_layer_norm(src, dst):
    src_ln = []