change adjustNodeShape to InitLayer, SaveModel and contnue now work