search

lkevinzc / dance

Codes for "DANCE: A Deep Attentive Contour Model for Efficient Instance Segmentation", WACV2021
67 stars 13 forks source link

Can I have a simple training on a graphics card？ #7

Closed David9591 closed 3 years ago

David9591 commented 3 years ago

First of all, thank you for your excellent work.I have successfully tested your model so far, but when I tried to train the model so that I could observe the flow of data, I ran into a problem: I only have one piece of 1080Ti(11G), and when I ran it, the video memory was exceeded. What parameters should I adjust if I want to train on a video card?Could you give me some advice?Thank you very much!

lkevinzc commented 3 years ago

@David9591 Hi there! Thanks and sorry for the late reply. If you have the out of memory error, you may try to use smaller backbone network, or you can reduce the input image resolution.