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

added option to change default tensor

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+2
View File
@@ -5,8 +5,10 @@ from pytorch_lightning.root_module.memory import get_gpu_memory_map
import traceback
from pytorch_lightning.root_module.model_saving import TrainerIO
from torch.optim.lr_scheduler import MultiStepLR
import pdb
try:
pdb.set_trace()
from apex import amp
APEX_AVAILABLE = True
except ModuleNotFoundError: