Open sun0717 opened 1 year ago
class C3STR(C3):
def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5): super().__init__(c1, c2, c2, n, shortcut, g, e) c_ = int(c2 * e) num_heads = c_ // 32 self.m = SwinTransformerBlock(c_, c_, num_heads, n)
super().init(c1, c2, c2, n, shortcut, g, e) <-- maybe there has two c2? and correct solution is super().init(c1, c2, n, shortcut, g, e) that model can trainning.
class C3STR(C3):
C3 module with SwinTransformerBlock()
super().init(c1, c2, c2, n, shortcut, g, e) <-- maybe there has two c2? and correct solution is super().init(c1, c2, n, shortcut, g, e) that model can trainning.