models-pytorch
models-pytorch copied to clipboard
Error in like 31 csp_densenet suggestion for fixing
def forward(self,x): features = [x[:,self.csp_num_features1:,...]] of line 31 stores features as a tuple because of the additional brackets this breaks the code you have to undo the bracket outside :)