Open divided7 opened 1 year ago
`class Model(nn.Module): ······ def forward(self, x, x_target, x_last, A_act, lamda_act): print("A_act.shape = {}, lamda_act = {}".format(A_act.shape)
A_act.shape=2,2,25,25 What's the A_act?
`class Model(nn.Module): ······ def forward(self, x, x_target, x_last, A_act, lamda_act): print("A_act.shape = {}, lamda_act = {}".format(A_act.shape)