分类：ShuffleNetv2

2021
06-09

Python深度学习之使用Pytorch搭建ShuffleNetv2

一、model.py1.1ChannelShuffledefchannel_shuffle(x:Tensor,groups:int)->Tensor:batch_size,num_channels,height,width=x.size()channels_per_group=num_channels//groups#reshape#[batch_size,num_channels,height,width]->[batch_size,groups,channels_per_group,height,width]x=x.view(batch_size,groups,channels_per_group,height,width)x=torch.transpose(x,... 继续阅读 >