Explicitly delete device-converted tensors.

Fixes dotchen#4.
tomcur · Apr 17, 2020 · 14daa4a · 14daa4a
1 parent c5422a5
commit 14daa4a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/training/train_birdview.py b/training/train_birdview.py
@@ -148,6 +148,9 @@ def train_or_eval(criterion, net, data, optim, is_train, config, is_first_epoch)
 
         tick = time.time()
 
+        del birdview, location, command, speed
+        del pred_location, loss, loss_mean
+
         if is_first_epoch and i == 10:
             iterator_tqdm.close()
             break

diff --git a/training/train_image_phase0.py b/training/train_image_phase0.py
@@ -204,6 +204,9 @@ def train_or_eval(coord_converter, criterion, net, teacher_net, data, optim, is_
 
         tick = time.time()
 
+        del rgb_image, birdview, location, command, speed
+        del _teac_location, _pred_location, pred_location, teac_location, loss, loss_mean
+
         if is_first_epoch and i == 10:
             iterator_tqdm.close()
             break

diff --git a/training/train_image_phase1.py b/training/train_image_phase1.py
@@ -224,6 +224,10 @@ def train_or_eval(coord_converter, criterion, net, teacher_net, data, optim, is_
 
         tick = time.time()
 
+        del rgb_image, birdview, location, command, speed
+        del _teac_location, _teac_locations, _pred_location, _pred_locations, pred_location, pred_locations
+        del loss, loss_mean
+
         if is_first_epoch and i == 10:
             iterator_tqdm.close()
             break