update: mouth helper

inference/tricks.py

@@ -65,7 +65,7 @@ class Trick(object):
 
     @staticmethod
     def arr_to_tensor(arr, norm: bool = True):
-        tensor = torch.tensor(arr, dtype=torch.float).cuda() / 255  # in [0,1]
+        tensor = torch.tensor(arr, dtype=torch.float).to(global_device) / 255  # in [0,1]
         tensor = (tensor - 0.5) / 0.5 if norm else tensor  # in [-1,1]
         tensor = tensor.permute(0, 3, 1, 2)
         return tensor