Merge pull request #15 from gsig/asynctf

asynctf updating api
gsig · May 7, 2019 · ded24bd · ded24bd
2 parents 7324e21 + ca789cb
commit ded24bd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tasks/video_task.py b/tasks/video_task.py
@@ -57,6 +57,7 @@ def validate_video(self, loader, model, criterion, epoch, args):
                       '{metrics}'.format(
                           i, len(loader), timer=timer, name=args.name, task=self.name,
                           metrics=' \t'.join(str(m) for m in metrics)))
+            del loss, output, target  # make sure we don't hold on to the graph
         submission_file(
             ids, outputs, '{}/epoch_{:03d}.txt'.format(args.cache, epoch+1))
         metrics = dict(m.compute() for m in metrics)