去雨去雪去雾算法本地实验速度对比

博主在去雨去雪去雾算法实验中,因原笔记本GTX960M速度慢,对GTX960M、RTX3070和RTX4060进行实验对比,记录不同显卡在不同batch - size下的显存占用和实验用时,还探讨了num_workers设置问题,发现不同显卡性能有差异。

在进行去雨去雪去雾算法的实验过程中,博主使用自己的电脑进行实验,尽管能够成功运行,但速度却属实不尽人意,博主的笔记本显卡是GTX960,显存为4G,在进行实验的过程中,batch-size只能设置为3,好在实验的去噪类型也就只与三种,否则这实验根本无法进行,于是,便萌生了更换笔记本的念头,那么在更换前,就要先做下相关实验,看看性能能够提升多少。

首先记录一下博主在GTX960M上的实验情况:

GTX960M实验情况

在这里插入图片描述

显存占用

在这里插入图片描述

实验用时

训练一个epoch的实验用时:

在这里插入图片描述
可以看到时间大约在15分钟左右。

每训练5个epoch进行一次评估:

在这里插入图片描述

属实是太慢了,用时31分钟,而且从结果来看,PSNR与SSIM竟然有些许差距,这才是最严重的,当然这并不是最终结果。

RTX3070实验情况

博主的实验配置为GTX3070(笔记本的哈),显存为8G。那么博主的batch-size就可以

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

彭祥.

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值