【7班】-MMPretrain代码与作业

MMPretrain环境搭建与配置 

安装指定分支的 MMEngine MMEngine new_config 环境搭建

git clone -b new_config_type https://github.com/HAOCHENYE/mmengine.git mmengine_new_config
cd mmengine_new_config
# 如果是新环境,没有安装 MMEngine 相关的依赖,需执行
pip install -e .

算法库适配

基于 new_config 分支,源码编译算法库

 源码编译 MMCV

注意: 

1. -b new_config 是为了切换支持mmengine的分支

2. 此处的“.[multimodal]”为添加多模态依赖,代表要安装多模态相关的额外依赖,

3.  MMCV_WITH_OPS=1 切换到和mmpretrain 配合使用的分支

4. 编译安装mmcv时会很慢,请耐心等待
 

config文件

resnet18_fintune.py

原始的resnet18_8xb32_in1k.py中__base__中提到的几个文件中的内容复制进去,后面会对这个文件修改或者作为继承文件

resnet18_fintuneM.py

修改复制esnet18_fintune.py模板中所需修改的参数

resnet18_fintuneM2.py  继承机制,简单明了
基于继承机制,继承修改自resnet18_fintune.py

注意:resnet18_fintuneM2.py 与resnet18_fintuneM.py 等价
 resnet18_fintuneM.py采用了完整的模板修改,

而resnet18_fintuneM2.py使用继承机制,只呈现了修改的关键参数,更加简单明了

修改要点
num_class=2    #对应数据集种类个数
init_cfg = dict(type='Pretrained', checkpoint='https://download.openmmlab.com/mmclassification/v0/resnet/resnet18_8xb32_in1k_20210831-fbbb1da6.pth')
checkpoint model weight 链接寻找链接和截图
欢迎来到 MMPretrain 中文教程! — MMPretrain 1.0.0rc8 文档

ResNet — MMPretrain 1.0.0rc8 文档

先建个档案 后面慢慢补充。。。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值