如何使用'；torch.utils.mobile#u优化器'；在PyTorch 1.6.0中？

根据（目前适用于1.6.0）

Torch mobile支持Torch.mobile\u optimizer.optimize\u，以便mobile utility在评估模式下运行带有模块的优化传递列表。该方法采用以下参数：torch.jit.ScriptModule对象、黑名单优化集和保留的方法列表

页面标题和

最后建议改为

torch.utils.mobile\u optimizer

。但是，这两种方法都不适用于我：

In [25]: module1 = torch.utils.mobile_optimizer.optimize_for_mobile(module)                                                                                
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-25-79e42fe53cdf> in <module>
----> 1 module1 = torch.utils.mobile_optimizer.optimize_for_mobile(module)

AttributeError: module 'torch.utils' has no attribute 'mobile_optimizer'

In [26]: module1 = torch.mobile_optimizer.optimize_for_mobile(module)                                                                                      
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-26-469dd450df3d> in <module>
----> 1 module1 = torch.mobile_optimizer.optimize_for_mobile(module)

AttributeError: module 'torch' has no attribute 'mobile_optimizer'

In [27]: print(torch.__version__)                                                                                                                          
1.6.0

[25]中的

module1=torch.utils.mobile\u optimizer.optimize\u for\u mobile（模块）
---------------------------------------------------------------------------
AttributeError回溯（最近一次呼叫上次）
在里面
---->1 module1=torch.utils.mobile\u optimizer.optimize\u for\u mobile（模块）
AttributeError:模块“torch.utils”没有属性“mobile\u optimizer”
[26]：module1=torch.mobile\u optimizer.optimize\u for\u mobile（模块）
---------------------------------------------------------------------------
AttributeError回溯（最近一次呼叫上次）
在里面
---->1模块1=火炬。移动\u优化器。为移动（模块）优化\u
AttributeError:模块“torch”没有属性“mobile\u optimizer”
在[27]中：打印（火炬版本）
1.6.0

我遗漏了什么吗？例如，我是否必须安装除

torch

本身以外的其他软件包

import torch.utils.mobile_optimizer as mobile_optimizer
scripted_model = torch.jit.script(model, pt_inputs)
opt_model = mobile_optimizer.optimize_for_mobile(scripted_model)
torch.jit.save(opt_model, "model.pt")

它对我有用

它对我有用

import torch.utils.mobile_optimizer as mobile_optimizer
scripted_model = torch.jit.script(model, pt_inputs)
opt_model = mobile_optimizer.optimize_for_mobile(scripted_model)
torch.jit.save(opt_model, "model.pt")