search

huyquoctrinh / MetaPolyp-CBMS2023

Meta-Polyp: a baseline for efficient Polyp segmentation (CBMS 2023)
MIT License
68 stars 9 forks source link

issue during training #6

Closed umang4002 closed 9 months ago

umang4002 commented 10 months ago

image

huyquoctrinh commented 10 months ago

image

Hi, this issue looks like you miss / in your directory, it should be TrainDataset/masks/. I hope you can fix it

umang4002 commented 10 months ago

Hi. Thankyou for the help. I have another doubt. Can you please tell that what are the GPU and ram specifications for running train.py?

umang4002 commented 10 months ago

And can it be run by decreasing the size on Nvidia T4 15 GB gpu ?

huyquoctrinh commented 10 months ago

And can it be run by decreasing the size on Nvidia T4 15 GB gpu ?

Hi @umang4002 , it can run on T4, but you should decrease the batch size, also maybe colab limits your resource, so in some case you should do tuning to prevent OOM. I hope it's ok, let me know what can I help

umang4002 commented 10 months ago

Hi. Thankyou for reaching out.

umang4002 commented 9 months ago

Thankyou huyquoctrinh the code was running fine on collab.

huyquoctrinh commented 9 months ago

Hi, it's my pleased to see it can run 😄