TypechoJoeTheme

鱼一的博客 ◡̈

yuyi

知不可乎骤得,托遗响于悲风
网站页面
标签搜索
文章目录

科研流水账30

1.将 pvasf 的项目在6005上运行,并使显卡负载均衡。
2.在 6005 上进行了融合三阶段输出特征和四阶段输出特征的实验。
3.找到了上周在 6002 上修改代码显存不够的主要原因。

VeRi Rank-1 Rank-5 Rank-10 mAP Epoch
1 - - -
2 96.66 98.57 99.23 82.2 129
3 - - - 81.7

问题分析

  • 在实现单机多卡负载均衡时遇到的错误
    修改 nn/parallel/scatter_gather.py 文件中 scatter 函数传入的第二个参数 chunk_sizes,默认None,会根据检测到的显卡设备数自动平均分配。一开始修改成 [48, 24],后面会报错,因为最后一个传入的batchsize 不一定为72。于是修改代码为根据输入 batchsize 来划分。

在 6002 上尝试复现之前跑过的最好结果,却报错显存不够

下步

目前在 6005 上可以用最好的模型在 vehicleID 数据集上进行训练,但是 6005 上的训练时间比6002 长1.5倍左右。
模型暂定为使用 stage3 和 stage4 输出的特征图进行融合,在 6005 上的实验相比最好结果只下降了0.1个点。
下一步计划是准备消融实验相关的内容。

赞(0)
版权属于:

鱼一的博客 ◡̈

本文链接:

https://yuyi.monster/archives/51/(转载时请注明本文出处及文章链接)

评论 (0)

More Info for me 📱

IP信息

人生倒计时

今日已经过去小时
这周已经过去
本月已经过去
今年已经过去个月