updates

2019-07-14 11:29:07 +02:00
parent 831b6e39b6
commit 3fc676b28a
2 changed files with 11 additions and 9 deletions
@@ -2,6 +2,7 @@ import torch


 def init_seeds(seed=0):
+    torch.cuda.empty_cache()
    torch.manual_seed(seed)
    torch.cuda.manual_seed(seed)
    torch.cuda.manual_seed_all(seed)