hellozhuo/pidinet

batch_size

long123524 opened this issue · 12 comments

long123524 commented

我有个疑问，batch_size为啥要设置成1呢？这样得到的loss比较大，但结果上看确是不错的

zhuoinoulu commented

我有个疑问，batch_size为啥要设置成1呢？这样得到的loss比较大，但结果上看确是不错的

loss和README表格里提供的training logs差别大吗?
batch size设置成1是 因为输入图片的分辨率不是统一的. 程序里边实际是用iter size来实现minibatch的功能, 每跑iter_size个样本就运行一次sgd.

long123524 commented

我设置成1时loss和你的差别很大，我的才1000多。当我把batch_size设置成其他值时，结果也还行。我的图片大小都是统一的，是不是增大batch_size更合适呢？

zhuoinoulu commented

我设置成1时loss和你的差别很大，我的才1000多。当我把batch_size设置成其他值时，结果也还行。我的图片大小都是统一的，是不是增大batch_size更合适呢？

所以是得到的loss比较小吗？可以增大batch size试试，如果大小统一，那么这个iter size可以设成1，不然实际的batch size就是batch_size x iter_size了

KengDong commented

我设置成1时loss和你的差别很大，我的才1000多。当我把batch_size设置成其他值时，结果也还行。我的图片大小都是统一的，是不是增大batch_size更合适呢？

你好，我在main.py中的batch_size更改為2，4，8之後就會出現這樣的錯誤。請問你那邊如何解決，我是想更改batch_size來增大我的GPU利用率
RuntimeError: expand(torch.cuda.FloatTensor{[2, 1, 480, 640]}, size=[480, 640]): the number of sizes provided (2) must be greater or equal to the number of dimensions in the tensor (4)

long123524 commented

我没有遇到这样的错误，你试试batch_aize为1时有没有出现这样的错误

emmmmmmmer commented

我没有遇到这样的错误，你试试batch_aize为1时有没有出现这样的错误

batch_size为1的时候是没有错的，请问你更改了batch_size后，GPU的占用率是否提升了，还有就是你测试的数据集是BSDS那种吗？因为我测试的数据集是480*640我在想是不是图片格式大小的问题。

long123524 commented

我测试的是自己的数据集，大小都是一样的256*256,你可以把图片大小统一试试，然后再加大batch。增大之后运行时间会减少许多

emmmmmmmer commented

我测试的是自己的数据集，大小都是一样的256256,你可以把图片大小统一试试，然后再加大batch。增大之后运行时间会减少许多
我的图片也是统一480640的，最后请问一下，你改动的batch_size是在main.py下的那个参数吗？

long123524 commented

是的，我也尝试了其他的batch_size

emmmmmmmer commented

是的，我也尝试了其他的batch_size

我刚试了一下，只测试一张图片就不会报错，但是测试2张图片就报错了，那边是测试多张图片都是可以的吗

long123524 commented

是的,你也可以自己写个测试代码

emmmmmmmer commented

是的,你也可以自己写个测试代码

好的，谢谢，我再试试能不能找到解决方案