HongHuang
Results
2
comments of
HongHuang
I found ``model = torch.nn.DataParallel(model, device_ids=device_ids)`` is necessary for pruning but I do not know the reason.
inner_dim and output channel needs to be a multiple of 4, e.g. 4, 8, 12, 16, ...