#positions = scale_tensor( # positions, (-self.cfg.radius, self.cfg.radius), (-1, 1) #) d08dc68 verified michaelj commited on Mar 29, 2024
v_pos.to(level.device), t_pos_idx.to(level.device) c851a4b verified michaelj commited on Mar 29, 2024