Exemplos de SLURMEnvironment em Python

Exemplo n.º 1

 def _select_environment(self):
     if self.trainer.plugin_connector.cloud_environment:
         env = self.trainer.plugin_connector.cloud_environment
     elif self.trainer.is_slurm_managing_tasks:
         env = SLURMEnvironment()
     elif self._is_using_torchelastic():
         env = TorchElasticEnvironment()
     else:
         env = TorchElasticEnvironment()
     return env

Exemplo n.º 2

Arquivo: accelerator_connector.py Projeto: youtang1993/pytorch-lightning

    def _select_environment(self):
        env = None

        # in priority: user environment, torchelastic (which is a generic environment), slurm
        if self.cluster_environment is not None:
            env = self.cluster_environment
        elif self._is_using_torchelastic():
            env = TorchElasticEnvironment()
        elif self.trainer.is_slurm_managing_tasks:
            env = SLURMEnvironment()
        return env