Giters

TuSimple / TuSimple-DUC

Understanding Convolution for Semantic Segmentation

Home Page:https://arxiv.org/abs/1702.08502

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How much GPU memory is required

amiltonwong opened this issue · comments

commented

Hi, authors,

How much GPU memory is required, in training and prediction, respectively?
Does a 6GB GPU card meet the requirement?

THX!

commented

It depends on your patch size and way of usage. In terms of training, a 320*320 patch with batch size 1 should take about 3GB. As for testing, a full resolution of 2048 * 1024 should take above 6GB GPU memory. So if you want to get better results, I would highly recommend you use a GPU with more than 8GB memory.