摘要:
class Mish(nn.Module): @staticmethod def forward(x): return x * F.softplus(x).tanh() class MemoryEfficientMish(nn.Module): class F(torch.autograd.Func 阅读全文
摘要:
仅使用nn.DataParallel,gpu0和gpu1、gpu0和gpu2、gpu0和gpu3等包含gpu0的组合都是可以的,其余组合不行,报错RuntimeError: module must have its parameters and buffers on device cuda:1 (d 阅读全文
摘要:
The Code of Pruning Filters For Efficient ConvNets 1. 代码参考 https://github.com/tyui592/Pruning_filters_for_efficient_convnets 其中主要是用VGG来进行在CIFAR100上的剪枝 阅读全文