db26c81
1
2
3
4
5
6
7
8
9
import torch import unittest class TestCase(unittest.TestCase): def tearDown(self): torch.cuda.empty_cache()
