updated training head detector