nn

Lightning-AI · Feb 1, 2021 · 1d9516c · 1d9516c
1 parent 0b44200
commit 1d9516c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -361,6 +361,7 @@ def package_list_from_file(file):
 import importlib
 import os
 import torch
+from torch import nn
 
 import pytorch_lightning as pl
 from pytorch_lightning import LightningModule, Trainer