Merge pull request advimman#163 from calpt/main

Don't set debug signal handler on Windows
IFF-0303 · Nov 18, 2023 · 5e921c0 · 5e921c0
2 parents ee58655 + b311798
commit 5e921c0
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/bin/predict.py b/bin/predict.py
@@ -38,7 +38,8 @@
 @hydra.main(config_path='../configs/prediction', config_name='default.yaml')
 def main(predict_config: OmegaConf):
     try:
-        register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
+        if sys.platform != 'win32':
+            register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
 
         device = torch.device(predict_config.device)
 

diff --git a/bin/predict_inner_features.py b/bin/predict_inner_features.py
@@ -38,7 +38,8 @@
 @hydra.main(config_path='../configs/prediction', config_name='default_inner_features.yaml')
 def main(predict_config: OmegaConf):
     try:
-        register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
+        if sys.platform != 'win32':
+            register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
 
         device = torch.device(predict_config.device)
 

diff --git a/bin/to_jit.py b/bin/to_jit.py
@@ -27,7 +27,8 @@ def forward(self, image, mask):
 
 @hydra.main(config_path="../configs/prediction", config_name="default.yaml")
 def main(predict_config: OmegaConf):
-    register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
+    if sys.platform != 'win32':
+        register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
 
     train_config_path = os.path.join(predict_config.model.path, "config.yaml")
     with open(train_config_path, "r") as f:

diff --git a/bin/train.py b/bin/train.py
@@ -31,7 +31,8 @@ def main(config: OmegaConf):
     try:
         need_set_deterministic = handle_deterministic_config(config)
 
-        register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
+        if sys.platform != 'win32':
+            register_debug_signal_handlers()  # kill -10 <pid> will result in traceback dumped into log
 
         is_in_ddp_subprocess = handle_ddp_parent_process()