wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-30 13:40:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleared spaces

Browse Source
This commit is contained in:
William Falcon committed 2020-01-21 14:02:54 -05:00
1 parent ef3fe53801
commit 22bd7dc920
1 file changed
-2
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/core/lightning.py
-2
View File
@@ -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