Available value list of Accelerator.barrier's name argument

[tarepan]

❓ Questions and Help

Are there the list of available Accelerator.barrier's name argument value?

What is your question?

Accelerator.barrier function accept name argument (doc)
Type of name is Optional[str], so there is no information of available str value.
Are there available value list?

By the way, in CheckpointConnector, name='TrainerIOMixin.restore_weights' is used.
There is no module named TrainerIOMixin, and restore_weights is hard coded function name (it is now too old name...).
If the list is clarified, we can refactor this not good name value.

Code

from doc:

class MyAccelerator(Accelerator):
  ...
  def barrier(self, name: Optional[str] = None):
            # implement this accelerator's barrier

name is Optional[str].

What's your environment?

master f740245

[carmocca]

The name argument is just for readability. As a comment for why the barrier is performed.

[tarepan]

Great info, thanks!