huiqu18 / FullNet-varCE

Pytorch implementation of FullNet-varCE

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hi, can we train fullnet with a larger input image size?

weiaicunzai opened this issue · comments

Thanks for your great work. I've tried to train fullnet using a larger input image size, e.g. 473x473, but always gives me cuda out of memory error, Im using a 3090 (24gb memory), I just want to make sure that if we can train fullnet with a much larger input image size like 512 x 512 or not, thanks.

@weiaicunzai The training of fullnet does require a lot of memory which is a major drawback. It is impossible to train with images with 512x512, unless you reduce the number of layers in each block.