kaixin96 / PANet

Code for our ICCV 2019 paper PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment
321 stars 64 forks source link

Any update plan about the Dataloaders and VRAM usage #19

Closed TimandXiyu closed 4 years ago

TimandXiyu commented 4 years ago

Hi, it has been very helpful to my on going project. However, is there any on-going plan that the dataloader could be simplified? Transfering the model to other dataset is really inconvenient. Another problem: increasing the batchsize dramatically increases the VRAM usage. Not sure whether this is normal or not. And if it is normal, is there anyway to do some optimization?

kaixin96 commented 4 years ago

@TimandXiyu Sorry for the late reply. I have a plan to reimplement the dataloader part using torchmeta but I am afraid I cannot give you a timeline. Torchmeta has a nice and cleaner interface for general few shot learning problems so reimplementing should not be difficult. You can try coding it yourself.

For the second question, I have not paid much attention to the VRAM usage before. I guess it is due to frequent IO calls. Maybe transferring to torchmeta can help.

kaixin96 commented 4 years ago

I’m closing this issue because it has been inactive for a while. Feel free to reopen if you have questions. Thank you.