src/datasets.py [370:378]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                     DatasetKeyConstants.image_sample_indices: random_sample_indices,
                     DatasetKeyConstants.image_pose: self.poses[index][None, :],
                     DatasetKeyConstants.image_rotation: self.rotations[index][None, :],
                     DatasetKeyConstants.ray_directions: self.directions}

        sample_dict = {}

        for feature_idx in range(len(self.train_config.f_in)):
            f_in = self.train_config.f_in[feature_idx]
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/datasets.py [465:473]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                     DatasetKeyConstants.image_sample_indices: random_sample_indices,
                     DatasetKeyConstants.image_pose: self.poses[index][None, :],
                     DatasetKeyConstants.image_rotation: self.rotations[index][None, :],
                     DatasetKeyConstants.ray_directions: self.directions}

        sample_dict = {}

        for feature_idx in range(len(self.train_config.f_in)):
            f_in = self.train_config.f_in[feature_idx]
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -