accelerate

bbcdbbaf - Remove checkpoints only on main process (#1974)

Remove checkpoints only on main process (#1974) * Remove checkpoints only on main process shutil.rmtree might throw errors if called on multiply processes. Make a call only on main process * Apply style