This commit is contained in:
Ftps
2023-11-19 04:28:56 +09:00
parent d1106fdd90
commit 59ddaacad9
3 changed files with 1 additions and 5 deletions
+1 -3
View File
@@ -135,9 +135,7 @@ def ipex_init():
torch.cuda.amp.GradScaler = torch.xpu.amp.GradScaler
except Exception:
try:
from .gradscaler import (
gradscaler_init,
)
from .gradscaler import gradscaler_init
gradscaler_init()
torch.cuda.amp.GradScaler = torch.xpu.amp.GradScaler