292cf321cb5f9ec96ac97d6ec9f94c271e9e8567,src/megaface/gen_megaface.py,,main,#Any#,111

Before Change

    net = edict()
    net.ctx = ctx
    net.sym, net.arg_params, net.aux_params = mx.model.load_checkpoint(prefix, epoch)
    net.arg_params, net.aux_params = ch_dev(net.arg_params, net.aux_params, net.ctx)
    all_layers = net.sym.get_internals()
    net.sym = all_layers["fc1_output"]
    //_pp = prefix.rfind("p")+1

After Change

    //net.arg_params, net.aux_params = ch_dev(net.arg_params, net.aux_params, net.ctx)
    all_layers = net.sym.get_internals()
    net.sym = all_layers["fc1_output"]
    net.model = mx.mod.Module(symbol=net.sym, context=net.ctx, label_names = None)
    net.model.bind(data_shapes=[("data", (1, 3, image_shape[1], image_shape[2]))])
    net.model.set_params(net.arg_params, net.aux_params)
    //_pp = prefix.rfind("p")+1
    //_pp = prefix[_pp:]
    //net.patch = [int(x) for x in _pp.split("_")]

Instances