update code for InferenceOptimizer

app.py

@@ -45,7 +45,7 @@ import time
 from data import PatchDataModule, prepare_data, image2tensor, tensor2image
 import torch
 from tqdm import tqdm
-from bigdl.nano.pytorch.trainer import Trainer
+from bigdl.nano.pytorch import InferenceOptimizer
 from torch.utils.data import DataLoader
 from pathlib import Path
 from torch.utils.data import Dataset
@@ -83,8 +83,9 @@ dm = PatchDataModule(train_image_dd, patch_size=2**6,
 # quantize model
 train_loader = dm.train_dataloader()
 train_loader_iter = iter(train_loader)
-quantized_model = Trainer.quantize(generator, accelerator=None,
-                                   calib_dataloader=train_loader)
+quantized_model = InferenceOptimizer.quantize(generator,
+                                              accelerator=None,
+                                              calib_dataloader=train_loader)
 
 
 def original_transfer(input_img):

requirements.txt

@@ -1,5 +1,5 @@
 # nano default
-bigdl-nano[pytorch]==2.1.0b20220606
+bigdl-nano[pytorch]==2.2.0b20221022
 setuptools==58.0.4
 protobuf==3.20.1