models-pytorch
models-pytorch copied to clipboard
Line 28 Mistake on dimensions
self.transition = _Csp_Transition(trans_in_features,trans_in_features//2) this code snippet replaced by csp_infeatures+trans_in_features because each block gets concatenated x+6dense= >csp transition -- x+12dense=>csp_transition
Csp_transition module is design for reduce feature dimensions