Code terminates without training while using accelerate

[muellerzr]

@jaydeepb

   accelerator = Accelerator()
    trainer = accelerator.prepare(Trainer(
        model=model,
        args=training_args,
        train_dataset=lm_datasets["train"],
        eval_dataset=lm_datasets["validation"],
    )) 
    trainer.train() 

First thing I see, don’t do this. Trainer uses Accelerate so no need to do any of this