gpleiss / efficient_densenet_pytorch

A memory-efficient implementation of DenseNets
MIT License
1.52k stars 327 forks source link

Fixed number of parameters for DenseNet-BC #67

Closed vinhtq115 closed 4 years ago

vinhtq115 commented 4 years ago

When using your implementation, I noticed that the number of parameters doesn't match the original paper when used with DenseNet-BC_250_24 and DenseNet-BC_190_40, only DenseNet-BC_100_12 matches. This modification changes the first convolutional output channels to twice the growth rate so that the number of parameters matches those in the paper.