trochscript模型初次运行过慢解决方法

在c++中首次使用torchscript的模型时会进行一些优化处理,导致首次运行时间过慢,可通过如下设置屏蔽优化处理,从而避免过慢问题

c++ 示例
#include <torch/script.h> // One-stop header.
#include <torch/torch.h>

torch::jit::getProfilingMode() = false;
torch::jit::getExecutorMode() = false;
torch::jit::setGraphExecutorOptimize(false);

std::shared_ptr<torch::jit::script::Module> module_ = std::make_shared<torch::jit::script::Module>(
            torch::jit::load(
                workspace +
                    "../superglue/models/model/SuperGlue_indoor.pt",
                device_));

python中对应命令
torch._C._jit_set_profiling_mode()

torch.jit.optimized_execution()

BASICSR_JIT是BasicSR库中的一个选项,用于启用即时编译(Just-in-Time Compilation)功能。根据提供的引用内容,有几种不同的安装方法可以启用BASICSR_JIT。 如果你使用的是CUDA,可以按照以下步骤安装BasicSR并启用BASICSR_JIT: 1. 打开终端或命令提示符,并执行以下命令: ``` set BASICSR_EXT=True pip install basicsr -vvv ``` 或者将上述命令分为两行执行: ``` set BASICSR_EXT=True pip install basicsr -vvv ``` 如果你没有CUDA,可以按照以下步骤安装BasicSR并启用BASICSR_JIT: 1. 打开终端或命令提示符,并执行以下命令: ``` pip install basicsr pip install facexlib ``` 2. 进入GFPGAN-master目录,并执行以下命令: ``` pip install -r requirements.txt python setup.py develop ``` 如果你还想修复无头像的背景图片,还需要执行以下命令: ``` pip install realesrgan ``` 综上所述,根据你的环境和需求,可以选择适合的安装方法来启用BASICSR_JIT。 #### 引用[.reference_title] - *1* *2* [python 学习笔记18 GFPGAN人脸(图片)修复](https://blog.csdn.net/wzhang1987/article/details/122042167)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [老照片修复](https://blog.csdn.net/qq_36226586/article/details/131640411)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值