eelxpeng's picture
fix weights bug
ee55b25