search

bfortuner / pytorch_tiramisu

FC-DenseNet in PyTorch for Semantic Segmentation
MIT License
306 stars 66 forks source link

The number of parameters #14

Open pyni opened 6 years ago

pyni commented 6 years ago

Hi, in the original paper, the parameters for FC-DenseNet56 is about 1.5 M But your code requires about 4000MB gpu to do a forward computation.

I wonder why there is a huge difference between these.