annealing and warm start for pq-training