models/dr_constant.py [174:188]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.precisions = NeuralPrecisions(self.n_species, config.params.n_hidden_decoder_precisions, 4)
        self.version = 1

    def initialize_state(self, theta, _treatments):
        n_batch = theta.get_n_batch()
        n_iwae = theta.get_n_samples()
        zero = torch.zeros([n_batch, n_iwae])
        x0 = torch.stack(
            [
                theta.init_x,
                theta.init_rfp,
                theta.init_yfp,
                theta.init_cfp,
                zero,
                zero,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models/prpr_constant.py [101:115]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.precisions = NeuralPrecisions(self.n_species, config.params.n_hidden_decoder_precisions, 4)
        self.version = 1

    def initialize_state(self, theta, _treatments):
        n_batch = theta.get_n_batch()
        n_iwae = theta.get_n_samples()
        zero = torch.zeros([n_batch, n_iwae])
        x0 = torch.stack(
            [
                theta.init_x,
                theta.init_rfp,
                theta.init_yfp,
                theta.init_cfp,
                zero,
                zero,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -