From f82dc3387721d5c51ede73d5cfcc977442faa6fb Mon Sep 17 00:00:00 2001 From: Jing Xu Date: Mon, 6 Jun 2022 16:33:18 +0900 Subject: [PATCH] update api usage example --- recipes_source/intel_extension_for_pytorch.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/recipes_source/intel_extension_for_pytorch.rst b/recipes_source/intel_extension_for_pytorch.rst index 8c25b62096d..83f0517e94e 100644 --- a/recipes_source/intel_extension_for_pytorch.rst +++ b/recipes_source/intel_extension_for_pytorch.rst @@ -98,7 +98,7 @@ Float32 model.set_state_dict(torch.load(PATH)) optimizer.set_state_dict(torch.load(PATH)) # Invoke optimize function against the model object and optimizer object - model, optimizer = ipex.optimize(model, optimizer, dtype=torch.float32) + model, optimizer = ipex.optimize(model, optimizer=optimizer) for images, label in train_loader(): # Setting memory_format to torch.channels_last could improve performance with 4D input data. This is optional. @@ -131,7 +131,7 @@ BFloat16 model.set_state_dict(torch.load(PATH)) optimizer.set_state_dict(torch.load(PATH)) # Invoke optimize function against the model object and optimizer object with data type set to torch.bfloat16 - model, optimizer = ipex.optimize(model, optimizer, dtype=torch.bfloat16) + model, optimizer = ipex.optimize(model, optimizer=optimizer, dtype=torch.bfloat16) for images, label in train_loader(): with torch.cpu.amp.autocast():