使用paddlehub如何指定模型的下载路径?打包起来比较困难。
收藏
使用paddlehub调用训练好的模型时:
第一次使用的话会去下载robert的权重的一个路径中
但是如果要把项目整个打包的话(从而让别人可以免去环境配置操作,直接使用),要怎么指定那个路径。
或者说要怎么不让它去下载那个就直接用鸭。
(尝试过用directory去指定,但貌似没有用)
这问题困扰好久了,希望大佬们能够解惑下,代码就是上面那个了。
GT
已解决
5#
回复于2021-07
在Python代码的最上方导入os模块,并用os模块设置环境变量HUB_HOME为"./hub_home"<-可自己命名,这样每次初始化的时候就会从这个目录里搜索,如果搜索到则不下载。即使这个文件夹换了位置/机器,由于"./hub_home"是相对路径,这个操作依旧可行。
1
收藏
请登录后评论
还有一个方法是指定环境变量HUB_HOME为一个特定的目录,这样每次下载之前会先检查这个目录下是否有存在的module文件,如果有就会跳过下载
求GT大哥给些案例或文档。主要问题是想把那个模型初始权重的路径指定为一个稳定点的路径方便项目打包点。 想要把那个初始权重下载下来后指定为那个路径,这样它才不用去下载。
在Python代码的最上方导入os模块,并用os模块设置环境变量HUB_HOME为"./hub_home"<-可自己命名,这样每次初始化的时候就会从这个目录里搜索,如果搜索到则不下载。即使这个文件夹换了位置/机器,由于"./hub_home"是相对路径,这个操作依旧可行。
os.environ['hub_home'] = './modules'
我是像上面这样设置的,然后把什么tar包,解压缩的包统统放到了当前项目的modlues路径下,结果还是不行啊。
"env": {"HUB_HOME": "./paddlehub"},