diff --git a/egs/imagenet/CLS/swin_transformer/train.py b/egs/imagenet/CLS/swin_transformer/train.py
index b88db1daf..14f760b6c 100755
--- a/egs/imagenet/CLS/swin_transformer/train.py
+++ b/egs/imagenet/CLS/swin_transformer/train.py
@@ -874,6 +874,10 @@ def run(rank, world_size, args):
             rank=rank,
         )
 
+        if params.print_diagnostics:
+            diagnostic.print_diagnostics()
+            break
+
         validate(
             params=params,
             model=model,
@@ -882,10 +886,6 @@ def run(rank, world_size, args):
             tb_writer=tb_writer,
         )
 
-        if params.print_diagnostics:
-            diagnostic.print_diagnostics()
-            break
-
         save_checkpoint(
             params=params,
             model=model,