def _torch_load(self, file):
    if ('cuda' in str(self.device)):
        return torch.load(file)
    else:
        return torch.load(file, map_location=(lambda storage, loc: storage))
def _torch_load(self, file):
    if ('cuda' in str(self.device)):
        return torch.load(file)
    else:
        return torch.load(file, map_location=(lambda storage, loc: storage))