Method store

classification/ema_deepspeed.py:66–75

Save the current parameters for restoring later. Args: model: A model that parameters will be stored

(self, model)

                        assert key not in self.m_name2s_name

    def store(self, model):
        """
        Save the current parameters for restoring later.
        Args:
          model: A model that parameters will be stored
        """
        with GatheredParameters(model.parameters()):
            if deepspeed.comm.get_rank() == 0:
                parameters = model.parameters()
                self.collected_params = [param.clone() for param in parameters]

    def restore(self, model):
        """