def yaml(self):
        return yaml.dump(self.serialized, default_flow_style=False,
                         encoding=None)
def tearDown(self):
        if hasattr(self, "rng_state"):
            torch.set_rng_state(self.rng_state)