search

prlz77 / ResNeXt.pytorch

Reproduces ResNet-V3 with pytorch
MIT License
505 stars 121 forks source link

GPU memory usage #22

Open whuhxb opened 2 years ago

whuhxb commented 2 years ago

Hi, compared to ResNet, what the GPU memory usage is for ResNeXt? Will this take more GPU memory? Thanks.

SarperYurttas commented 2 years ago

Hi, compared to ResNet, what the GPU memory usage is for ResNeXt? Will this take more GPU memory? Thanks.

@whuhxb As the paper said, these two models have similar numbers of parameters. Hence GPU memory usage must have similar as well if it's implemented correctly.

resnext