Added freezing and unfreezing.
Refactored training regimes.