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 :)
the list type is for concat Muti features in line 36. And the input of DenseLayer can be a list type.