24bf6ad607b491fc28c83b930e7cec927599d73f,python/ray/util/sgd/torch/torch_runner.py,TorchRunner,setup,#TorchRunner#,139

Before Change

        self._create_schedulers_if_available()
        self._try_setup_apex()
        self._create_loss()
        self._initialize_dataloaders()
        self.training_operator = self.training_operator_cls(
            self.config,
            models=self.models,
            optimizers=self.optimizers,

After Change

    def setup(self):
        Initializes the model.
        _remind_gpu_usage(self.use_gpu, is_chief=True)
        self._initialize_dataloaders()
        logger.debug("Creating model")
        self.models = self.model_creator(self.config)
        if not isinstance(self.models, collections.Iterable):
            self.models = [self.models]

Instances