debug
weight grad fix