EX 5: name 'train_task_function' is not defined

when calling train_task = train_task_function(train_batch_stream); There is no definition of train_task_function in the notebook.

Maybe you are not running the cell which defines this function? Otherwise why it should not reckognise this function…?