Skip to content

Commit 5b7ce58

Browse files
committed
empty cuda cache
1 parent 535bcfb commit 5b7ce58

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

conf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def reset_seeds(gallery_conf, fname):
9292
torch.cuda.empty_cache()
9393
torch.manual_seed(42)
9494
torch.set_default_device(None)
95+
torch.cuda.empty_cache()
9596
random.seed(10)
9697
numpy.random.seed(10)
9798
gc.collect()

0 commit comments

Comments
 (0)