预训练模型还要训练吗_多模态预训练模型

预训练模型还要训练吗_多模态预训练模型若使用已保存好的镜像reid_mgn:v1,在本机上可按如下操作训练#1.进入已保存环境的镜像(reid_mgn:v1(8.48G)、pytorch/pytorch:1.0.1-cuda10.0-cudnn7-devel_mgnreid(6.37G))nvidia-dockerrun-it–rm-v/home/lc-deep/sdr:/home/personReID…

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE稳定放心使用

若使用已保存好的镜像reid_mgn:v1,在本机上可按如下操作训练

 

# 1.进入已保存环境的镜像(reid_mgn:v1(8.48G)、pytorch/pytorch:1.0.1-cuda10.0-cudnn7-devel_mgnreid(6.37G))

nvidia-docker run -it –rm  -v  /home/lc-deep/sdr:/home/personReID reid_mgn:v1 /bin/bash   (服务器10.10.113.130

 

nvidia-docker run -it –rm –name sxue -v  /home/lc-deep/sdr:/home/personReID pytorch/pytorch:1.0.1-cuda10.0-cudnn7-devel_mgnreid /bin/bash

 

docker attach ID

 

nvidia-docker run -it –rm –name mgn -v /home/deep/share2/docker_data:/home -v /home/deep/sxue:/home/personReID ufoym/deepo:testv1 /bin/bash    (75服务器)

 

# 2.进入到工程目录

cd /home/personReID/MGN-pytorch-master

 

 

# 3.复制预训练模型到指定路径(可跳过这步)

cp /home/personReID/MGN-pytorch-master/resnet50-19c8e357.pth /root/.cache/torch/checkpoints/resnet50-19c8e357.pth

#### 注每次需查电脑自动保存的根目录 /root/.cache/torch/checkpoints/resnet50-19c8e357.pth 会因电脑不同而不同

cp /home/personReID/MGN-pytorch-master/resnet50-19c8e357.pth /root/.torch/models/resnet50-19c8e357.pth

 

# 打开另一个终端

docker ps              查看容器内镜像(找到reid_mgn:v1 前对应的数字字符串%%%%)

docker stats %%%%%     实时监测内存情况

 

# 4.训练(在原终端继续进行,注:demo.sh是已改好参数的)

sh demo1.sh

 

 

补充: 训练前需要修改的文件及代码

 

1.demo.sh文件

修改data路径(把你的数据集路径添加到 –datadir)、GPU、参数等如下:

 

CUDA_VISIBLE_DEVICES=0 python main.py –reset –datadir /home/personReID/MGN-pytorch-master/Market1501/ –batchid 4 –batchtest 16 –test_every 20 –epochs 80 –decay_type step_40_60 –loss 1*CrossEntropy+2*Triplet –margin 0.3 –re_rank –random_erasing –save MGN_adam –nGPU 1  –lr 2e-4 –optimizer ADAM –save_models

注:需将数据集文件名由原始的Market-1501-****改为和代码匹配的Market1501

 

2.trainer.py

修改train、test中的epoch

预训练模型还要训练吗_多模态预训练模型

预训练模型还要训练吗_多模态预训练模型

3.main.py

如果是单GPU训练,需按如下添加命令

预训练模型还要训练吗_多模态预训练模型

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/185472.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)
blank

相关推荐

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号