pytorch_semantic_human_matting icon indicating copy to clipboard operation
pytorch_semantic_human_matting copied to clipboard

I feel it is slow when training.

Open xuehuachunsheng opened this issue 6 years ago • 3 comments

When I train the TNet, I feel frustrated because the train speed is too slow. I use 'aisegmentcom-matting-human-datasets' to train this model (the number of images is about 34000), batch_size=8, nearly 4400 batches per one epoch. Totally 300 epochs may train about 15 days. Is it normal? Or actually 300 epochs is no necessary?

xuehuachunsheng avatar Sep 18 '19 08:09 xuehuachunsheng

Single GPU using 1080Ti。

xuehuachunsheng avatar Sep 18 '19 08:09 xuehuachunsheng

what's the resolution of your input? usually the model tends to converge after 50 epochs

tsing90 avatar Sep 19 '19 16:09 tsing90

I use 800x600 HxW size of image to train the model's T-Net stage, but the trimap seems not too good. Is it normal?

xuehuachunsheng avatar Sep 23 '19 12:09 xuehuachunsheng