5c5a490629aa20568434d1c6d11f4be4b8155518,tests/convert_gl2tf2_lstm.py,,gl_calc,#,22

Before Change

    ctx = mx.gpu(0)
    gl_params = gl_model._collect_params_with_prefix()
    // gl_w = np.transpose(tf2_w, axes=(3, 2, 0, 1))
    gl_params["conv.weight"]._load_init(mx.nd.array(gl_w, ctx), ctx)
    // gl_params["conv.bias"]._load_init(mx.nd.array(b, ctx), ctx)

    gl_x = mx.nd.array(x, ctx)

After Change

    // gl_y = gl_model(gl_x).asnumpy()

    ctx = mx.gpu(0)
    gl_x = mx.nd.zeros((3, 7, 80), ctx)
    gl_model.initialize(ctx=ctx)
    gl_model(gl_x)
    gl_params = gl_model._collect_params_with_prefix()
    _calc_width(gl_model)

Instances