How about the GPU memory cost?

longcw / RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch

555 stars 103 forks source link

How about the GPU memory cost? #6

Open sshaoshuai opened 6 years ago

sshaoshuai commented 6 years ago

Hi, thanks for your sharing codes. So how about the GPU memory cost compared with original RoiPool? And it seems like the function grid_sample in PyTorch could also do the similar work, so what's the difference?