affordance_seg/collect_dset.py [207:222]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                if not_done_masks[i].item() == 0:

                    episode_stats = dict()
                    episode_stats["reward"] = current_episode_reward[i].item()
                    episode_stats.update(
                        self._extract_scalars_from_info(infos[i])
                    )
                    current_episode_reward[i] = 0

                    # use scene_id + episode_id as unique id for storing stats
                    stats_episodes[
                        (
                            current_episodes[i]['scene_id'],
                            current_episodes[i]['episode_id'],
                        )
                    ] = episode_stats
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



interaction_exploration/viz_trainer.py [202:214]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    if not_done_masks[i].item() == 0:
                        episode_stats = dict()
                        episode_stats["reward"] = current_episode_reward[i].item()
                        episode_stats.update(
                            self._extract_scalars_from_info(infos[i])
                        )
                        current_episode_reward[i] = 0
                        stats_episodes[
                            (
                                current_episodes[i]['scene_id'],
                                current_episodes[i]['episode_id'],
                            )
                        ] = episode_stats
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -