77e91e8bc57cefadc8607a0866ab51ba33b01762,src/sdk/pynni/nni/nas/pytorch/mutator.py,Mutator,export,#Mutator#,11

Before Change

    def export(self):
        if self._in_forward_pass:
            raise RuntimeError("Still in forward pass. Exporting might induce incompleteness.")
        if not self._cache:
            raise RuntimeError("No running history found. You need to call your model at least once before exporting. "
                               "You might also want to check if there are no valid mutables in your model.")
        return self._cache

    @contextmanager
    def forward_pass(self):

After Change

        -------
        dict
        
        return self.sample_final()

    def on_forward_layer_choice(self, mutable, *inputs):
        
        On default, this method calls :meth:`on_calc_layer_choice_mask` to get a mask on how to choose between layers

Instances