diff --git a/docs/conf.py b/docs/conf.py
index 311f774..44ffd72 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -20,7 +20,18 @@ class Mock(MagicMock):
         return MagicMock()
 
 
-MOCK_MODULES = ["numpy", "pyswip", "torch", "torchvision", "zoopt", "termcolor"]
+MOCK_MODULES = [
+    "numpy",
+    "pyswip",
+    "torch",
+    "torchvision",
+    "zoopt",
+    "termcolor",
+    "torch.nn",
+    "torch.utils",
+    "torch.optim",
+    "torchvision.transform",
+]
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)