From 8649ae65b82b2c324f98b57623295b3ab083da1a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 13:36:48 -0500 Subject: [PATCH] working on trainer docs --- pytorch_lightning/core/lightning.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 4a67ba56..ad684166 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -9,12 +9,13 @@ from argparse import Namespace import pandas as pd import torch import torch.distributed as dist - +# from pytorch_lightning.core.decorators import data_loader from pytorch_lightning.core.grads import GradInformation from pytorch_lightning.core.hooks import ModelHooks -from pytorch_lightning.core.memory import ModelSummary from pytorch_lightning.core.saving import ModelIO +from pytorch_lightning.core.memory import ModelSummary +from pytorch_lightning.trainer.training_io import load_hparams_from_tags_csv from pytorch_lightning.overrides.data_parallel import LightningDistributedDataParallel