def __init__(self, filename=None):
        self.filename = None
        if filename:
            self.load(filename)
def tearDown(self):
        if hasattr(self, "rng_state"):
            torch.set_rng_state(self.rng_state)