深度学习技巧

本文探讨了在训练深度学习模型时,如何选择合适的batch_size以避免内存溢出。通常,batch_size建议设置为2的幂,如2或4。对于unet模型,batch_size=48可能适合,但对较大的unet++模型,需要降低到4才能成功加载。此外,图像通常被裁剪为256*256、512*512或224*224等标准尺寸进行训练。
摘要由CSDN通过智能技术生成

1.batchsize不要太大,容易爆内存,一般就是4以下,比如2和4(2的n次方)

48就太大了(unet模型比较小,所以batch_size=48是可以的不会溢出内存,但是unet++模型比较大,所以batch_size=48就会溢出内存,实验时,我从48一直降到4才能将模型读入内存)

2.图片一般要裁剪成256*256或者是512*512又或者是224*224

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值