mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-30 13:40:46 +08:00
fixing Win failed import (#1163)
* version * try fix distrib * update try import
This commit is contained in:
1 parent
49d000c0c9
commit
e461ec0037
8 files changed
+18
-19
No files matched your search
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user