pytorch increase accuracy