    def get_snapshot(self, epoch):
        """Return loggable snapshot of the SQL algorithm.

        If `self._save_full_state == True`, returns snapshot including the
        replay pool. If `self._save_full_state == False`, returns snapshot
        of policy, Q-function, and environment instances.
        """

        state = {
            'epoch': epoch,
            'policy': self._policy,
            'Q': self._Q,
            'training_environment': self._training_environment,
            'evaluation_environment': self._evaluation_environment,
        }

        if self._save_full_state:
            state.update({'replay_pool': self._pool})

        return state
Return loggable snapshot of the SQL algorithm.

If `self._save_full_state == True`, returns snapshot including the
replay pool. If `self._save_full_state == False`, returns snapshot
of policy, Q-function, and environment instances.
softlearning/algorithms/sql.py