How do get I generate code that goes in a training_function() to be used with accelerator in notebook ?
training_function()