fixed trainer strong parameters issue