wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleared spaces

Browse Source
This commit is contained in:
William Falcon
2020-01-21 14:02:54 -05:00
parent ef3fe53801
commit 22bd7dc920
1 changed files with 0 additions and 2 deletions
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