Revert "Set training mode in master"

This reverts commit 6383b15cce61469a00f391c6c677fa9b191132c0.