search

iamhankai / ghostnet.pytorch

[CVPR2020] GhostNet: More Features from Cheap Operations
https://arxiv.org/abs/1911.11907
522 stars 116 forks source link

Replace nested nn.Sequential() with nn.Identity() #22

Closed glenn-jocher closed 4 years ago

glenn-jocher commented 4 years ago

Identity() should be faster with less overhead if the intention is for these modules to be simple pass-throughs. https://pytorch.org/docs/stable/nn.html#torch.nn.Identity

iamhankai commented 4 years ago

module 'torch.nn' has no attribute 'Identity' in Pytorch-1.0

glenn-jocher commented 4 years ago

@iamhankai ah! Of course, I see.