Make non_blocking optional for model trainers.
Make non_blocking optional for model trainers.