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

fixing Win failed import (#1163)

Browse Source 
* version

* try fix distrib

* update try import
This commit is contained in:
Jirka Borovec
2020-03-16 20:50:36 -04:00
committed by GitHub
parent 49d000c0c9
commit e461ec0037
8 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/core/hooks.py
+2 -2
View File
@@ -22,10 +22,10 @@ from torch.optim.optimizer import Optimizer
try:
from apex import amp
APEX_AVAILABLE = True
except ImportError:
APEX_AVAILABLE = False
else:
APEX_AVAILABLE = True
class ModelHooks(torch.nn.Module):