update seg inference

inference/seg.py

@@ -16,6 +16,9 @@ def load_model(task, version):
             return None, None
         
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        if torch.cuda.is_available() and torch.cuda.get_device_properties(0).major >= 8:
+            torch.backends.cuda.matmul.allow_tf32 = True
+            torch.backends.cudnn.allow_tf32 = True
         model = torch.jit.load(model_path)
         model.eval()
         model.to(device)