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
View File
@@ -4,7 +4,6 @@ import intel_extension_for_pytorch as ipex
import intel_extension_for_pytorch._C as core
import torch
OptState = ipex.cpu.autocast._grad_scaler.OptState
_MultiDeviceReplicator = ipex.cpu.autocast._grad_scaler._MultiDeviceReplicator
_refresh_per_optimizer_state = (