From 73f4b494572b4b4e7c55047147211b83585bdcd6 Mon Sep 17 00:00:00 2001
From: Daniel Manzke <daniel.manzke@googlemail.com>
Date: Mon, 3 Jan 2022 20:46:20 +0100
Subject: [PATCH] removed unnecessary flag

---
 demo/image_matting/colab/inference.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/demo/image_matting/colab/inference.py b/demo/image_matting/colab/inference.py
index d7423cf..f9a15e0 100644
--- a/demo/image_matting/colab/inference.py
+++ b/demo/image_matting/colab/inference.py
@@ -45,12 +45,11 @@ if __name__ == '__main__':
     # create MODNet and load the pre-trained ckpt
     modnet = MODNet(backbone_pretrained=False)
     modnet = nn.DataParallel(modnet)
-    cpu_mode = False
+
     if torch.cuda.is_available():
         modnet = modnet.cuda()
         weights = torch.load(args.ckpt_path)
     else:
-        cpu_mode = True
         weights = torch.load(args.ckpt_path, map_location=torch.device('cpu'))
     modnet.load_state_dict(weights)
     modnet.eval()