Trainer and Accelerate

related: Trainers.train() with accelerate