Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added option to change default tensor

This commit is contained in:
William Falcon 2019-05-13 20:44:25 -04:00
parent 7002de1d4e
commit 88fbf6cc4b
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/models/trainer.py
View File

@ -5,8 +5,10 @@ from pytorch_lightning.root_module.memory import get_gpu_memory_map
import traceback import traceback
from pytorch_lightning.root_module.model_saving import TrainerIO from pytorch_lightning.root_module.model_saving import TrainerIO
from torch.optim.lr_scheduler import MultiStepLR from torch.optim.lr_scheduler import MultiStepLR
import pdb
try: try:
pdb.set_trace()
from apex import amp from apex import amp
APEX_AVAILABLE = True APEX_AVAILABLE = True
except ModuleNotFoundError: except ModuleNotFoundError: