search

Knight825 / models-pytorch

This is a torchvision style CNN models collection based on pytorch.
Apache License 2.0
21 stars 5 forks source link

Error in like 31 csp_densenet suggestion for fixing #5

Open gurber15 opened 1 year ago

gurber15 commented 1 year ago

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 :)

Knight825 commented 1 year ago

the list type is for concat Muti features in line 36. And the input of DenseLayer can be a list type.