Can you tell me why aren’t you satisfied with using .training in forward method? It is really simple to implement and obviously follows convention. I cannot think of any situation that this cannot handle.