From 1d9516c9ec82a1f538d879e37d898b41142e61a4 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Mon, 1 Feb 2021 12:24:30 +0100 Subject: [PATCH] nn --- docs/source/conf.py | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/source/conf.py b/docs/source/conf.py index 3e3acbd3a5b64..42b02828348ad 100644 --- 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