diff --git a/auto3dseg/algorithm_templates/dints/scripts/search.py b/auto3dseg/algorithm_templates/dints/scripts/search.py
index c0357527..5079c486 100644
--- a/auto3dseg/algorithm_templates/dints/scripts/search.py
+++ b/auto3dseg/algorithm_templates/dints/scripts/search.py
@@ -40,10 +40,7 @@
 from monai.metrics import compute_dice
 from monai.utils import RankFilter, set_determinism
 
-try:
-    from apex.contrib.clip_grad import clip_grad_norm_
-except ModuleNotFoundError:
-    from torch.nn.utils import clip_grad_norm_
+from torch.nn.utils import clip_grad_norm_
 
 
 CONFIG = {
diff --git a/auto3dseg/algorithm_templates/dints/scripts/train.py b/auto3dseg/algorithm_templates/dints/scripts/train.py
index 9d0e36c1..fb996b6f 100644
--- a/auto3dseg/algorithm_templates/dints/scripts/train.py
+++ b/auto3dseg/algorithm_templates/dints/scripts/train.py
@@ -47,10 +47,7 @@
 from torch.utils.tensorboard import SummaryWriter
 from tqdm import tqdm
 
-try:
-    from apex.contrib.clip_grad import clip_grad_norm_
-except ModuleNotFoundError:
-    from torch.nn.utils import clip_grad_norm_
+from torch.nn.utils import clip_grad_norm_
 
 try:
     _libcudart = ctypes.CDLL("libcudart.so")