From 22bd7dc9209a9eae269209684438b0e57c4719fa Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 16:08:03 -0500 Subject: [PATCH] cleared spaces --- pytorch_lightning/core/lightning.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 93689ac2..20c62cf8 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1015,7 +1015,6 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): map_location (dict): A dictionary mapping saved weight GPU devices to new GPU devices (example: {'cuda:1':'cuda:0'}) - Return: LightningModule with loaded weights @@ -1136,7 +1135,6 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): def freeze(self): r""" - Freeze all params for inference Example