(self, extra_cond)
| 59 | setattr(_module, '_forward', bound_method) |
| 60 | |
| 61 | def get_traj_features(self, extra_cond): |
| 62 | b, c, t, h, w = extra_cond.shape |
| 63 | ## process in 2D manner |
| 64 | extra_cond = rearrange(extra_cond, 'b c t h w -> (b t) c h w') |
| 65 | traj_features = self.omcm(extra_cond) |
| 66 | traj_features = [rearrange(feature, '(b t) c h w -> b c t h w', b=b, t=t) for feature in traj_features] |
| 67 | return traj_features |
no outgoing calls
no test coverage detected