add test handler

2020-08-24 22:46:36 +08:00
parent 31aafb3470
commit 56b355737f
5 changed files with 65 additions and 9 deletions
--- a/util/handler.py
+++ b/util/handler.py
@@ -17,7 +17,7 @@ def empty_cuda_cache(_):


 def setup_common_handlers(trainer: Engine, config, stop_on_nan=True, clear_cuda_cache=False, use_profiler=True,
-                          to_save=None, end_event=None, set_epoch_for_dist_sampler=True):
+                          to_save=None, end_event=None, set_epoch_for_dist_sampler=False):
    """
    Helper method to setup trainer with common handlers.
    1. TerminateOnNan