detect_steel_bar icon indicating copy to clipboard operation
detect_steel_bar copied to clipboard

batch_size and collater

Open junliang230 opened this issue 4 years ago • 0 comments

thanks for your work. when I try to figure out it, I have some probloms about this. could you help me to explain these? 1.why the bath_size is set to 1, it seems too small 2.it seems to spend too much time in collater process when loading train data. so the GPU is waiting this GPU-Util is very slow . do you have good idea to solve it?

junliang230 avatar Oct 18 '19 13:10 junliang230