Я використовував trunc_normal_ для ваги на початку... 😔