(self, in_channels,num_slots_N,num_slots_M,iters,resolutions)
| 505 | |
| 506 | class ODE2(nn.Module): |
| 507 | def __init__(self, in_channels,num_slots_N,num_slots_M,iters,resolutions): |
| 508 | super(ODE2, self).__init__() |
| 509 | self.SAn = SlotAttentionModule2(in_channels, resolutions, num_slots_N, iters) |
| 510 | self.SAm = SlotAttentionModule2(in_channels, resolutions, num_slots_M, iters) |
| 511 | self.getalpha = getAlpha(in_channels) |
| 512 | |
| 513 | def forward(self, feature_map): |
| 514 | recon_combined1,f1 = self.SAn(feature_map) |
nothing calls this directly
no test coverage detected