fix bug in nv_load_mlp
refactor