From 3fae45c8c42feb64f0bdac63b809ede90ac0ed8d Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 09:26:35 -0500 Subject: [PATCH] working on trainer docs --- pytorch_lightning/trainer/trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 57855128..758de820 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -3,6 +3,7 @@ import sys import warnings import logging + import torch import torch.distributed as dist import torch.multiprocessing as mp