An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
I uesd the Partial_conv in my model, which includes the Channel Split(torch.split). In the inference(after ModelSpeedup), i got the error in x1, x2 = torch.split(x, [self.dim_conv3, self.dim_untouched], dim=1):
I uesd the Partial_conv in my model, which includes the Channel Split(torch.split). In the inference(after ModelSpeedup), i got the error in
x1, x2 = torch.split(x, [self.dim_conv3, self.dim_untouched], dim=1)
:RuntimeError: start (36) + length (108) exceeds dimension size (139).