friendly errors for issue pytorch#2207

DebarshiChanda · Oct 2, 2021 · 41e9d8d · 41e9d8d
1 parent eefd033
commit 41e9d8d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/examples/contrib/cifar10/main.py b/examples/contrib/cifar10/main.py
@@ -187,6 +187,9 @@ def run(
         **spawn_kwargs: Other kwargs to spawn run in child processes: master_addr, master_port, node_rank, nnodes
 
     """
+    # raise error if num_warmup_epochs is less than num_epochs
+    assert num_epochs >= num_warmup_epochs, 'num_epochs is less than num_warmup_epochs, Consider increasing num_epochs aor decreasing num_warmup_epochs'
+
     # catch all local parameters
     config = locals()
     config.update(config["spawn_kwargs"])

diff --git a/examples/contrib/cifar10_qat/main.py b/examples/contrib/cifar10_qat/main.py
@@ -174,6 +174,9 @@ def run(
         **spawn_kwargs: Other kwargs to spawn run in child processes: master_addr, master_port, node_rank, nnodes
 
     """
+    # raise error if num_warmup_epochs is less than num_epochs
+    assert num_epochs >= num_warmup_epochs, 'num_epochs is less than num_warmup_epochs, Consider increasing num_epochs aor decreasing num_warmup_epochs'
+
     # catch all local parameters
     config = locals()
     config.update(config["spawn_kwargs"])

diff --git a/examples/contrib/transformers/main.py b/examples/contrib/transformers/main.py
@@ -192,6 +192,9 @@ def run(
         with_amp (bool): if True, enables native automatic mixed precision. Default, False.
         **spawn_kwargs: Other kwargs to spawn run in child processes: master_addr, master_port, node_rank, nnodes
     """
+    # raise error if num_warmup_epochs is less than num_epochs
+    assert num_epochs >= num_warmup_epochs, 'num_epochs is less than num_warmup_epochs, Consider increasing num_epochs aor decreasing num_warmup_epochs'
+
     # catch all local parameters
     config = locals()
     config.update(config["spawn_kwargs"])