Policy training

shape