search

yinghdb / EmbedMask

EmbedMask: Embedding Coupling for One-stage Instance Segmentation
MIT License
160 stars 19 forks source link

Train Speed #1

Closed RainHxj closed 4 years ago

RainHxj commented 4 years ago

Does it take 4 days to train a model? （Titan RTX，4 gpu ，batchsize 16， （800，1333））

yinghdb commented 4 years ago

I have not trained on Titan RTX, but I can give the training time on Tesla P40.

It takes about 40 hours to train a model with the config of 'embed_mask_R50_1x.yaml' on 4 GPUs of Tesla P40.