logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Pytorch 加载多gpu模型+解决加载模型测试结果不正确

将模型加载到多gpu上的时候,会用到model = torch.nn.DataParallel(model, device_ids=[1, 2, 3])因此在保存模型的时候,若直接torch.save(model.state_dict(), save_dict_path)加载后的模型参数每个都会带有 module如果使用strict=False的加载方式,很有可能会使测试结果不同model.loa

#pytorch#深度学习#python
Python 循环写入json文件 解决内容覆盖+换行问题

一般使用 open打开一个json文件为文件标识符,使用文件标识符来对文件进行写入:import json# save = dict() 为待保存的字典with open("./res_video.json", 'w', encoding='utf-8') as fw:json.dump(save, fw, indent=4, ensure_ascii=False)json.dump会保存字典 s

#python
Python 读取 .gz 文件

读取.gz 文件需要使用gzip 包,如果没有安装可以自行在终端安装pip install gzipimport gzippath = "" #你的文件路径f = gzip.open(path, 'rb')for line in f.readlines(): # 按行进行读取s = line.decode() # 读取之后要进行解码print(s) # s 为string类型,就是我们读取的文件中

#python#开发语言
到底了