search

BarCodeReader / RCSB-PyTorch

[WACV2022]-Recursive Contour-Saliency Blending Network for Accurate Salient Object Detection
Apache License 2.0
39 stars 5 forks source link

training is slow #1

Closed ws0352 closed 2 years ago

ws0352 commented 2 years ago

image This is my GPU info, but it's slow when training. image

BarCodeReader commented 2 years ago

Hi, thanks for your interest in our work.

unfortunately, the training is slow, with 3090 we need a few days, it is because of the recurrent loop...

But as you can see from the evaluation results, the model converges very fast, in your case, the first epoch already get MAE 0.034 on DUTSTE...

so you can reduce the epochs and in the meantime, we are about to optimize the code, and you are also welcome to contribute by submitting PRs.

cheers

ws0352 commented 2 years ago

Thank you very much for your answer, I will continue to pay attention to your work, if necessary, I will contribute my part of the code.