请问模型训练时batch_size只能是1么？超过1会如何？

backseason / PoolNet

Code for our CVPR 2019 paper "A Simple Pooling-Based Design for Real-Time Salient Object Detection"

MIT License

628 stars 153 forks source link

Closed pengqianli closed 4 years ago

backseason commented 4 years ago

dataloader使用的图片为原始大小，所以batch_size只能为1。

pengqianli commented 4 years ago

感谢您的回复，也就是说，是显存大小限制了batchsize的大小了？（或者是因为考虑输入图片可能存在尺度不一样的情况？），不够怎么样，感谢答疑！

backseason commented 4 years ago

不是因为显存，只是为了保持测试时候可以接受任意尺寸的图片输入。当输入图片的尺寸不一样时，无法cat为一个tensor。

pengqianli commented 4 years ago

明白了，谢谢您！