def __init__(self, path):
        self.path = path
        self.body_labels = collections.OrderedDict()
def tearDown(self):
        if hasattr(self, "rng_state"):
            torch.set_rng_state(self.rng_state)