Fixed bug in hybrid training, removed gradient validation check (more appropriate for a different script)