06945cbd6dffb01a07af4239cf5c1753c8646ec9,reagent/training/world_model/mdnrnn_trainer.py,MDNRNNTrainer,__init__,#MDNRNNTrainer#Any#Any#Any#,23

Before Change

    ):
        self.memory_network = memory_network
        self.params = params
        self.optimizer = torch.optim.Adam(
            self.memory_network.mdnrnn.parameters(), lr=params.learning_rate
        )
        self.minibatch = 0
        self.minibatch_size = params.minibatch_size
        self.cum_loss: Deque[float] = deque([], maxlen=cum_loss_hist)
        self.cum_bce: Deque[float] = deque([], maxlen=cum_loss_hist)

After Change

        params: MDNRNNTrainerParameters,
        cum_loss_hist: int = 100,
    ):
        super().__init__()
        self.memory_network = memory_network
        self.params = params

    def configure_optimizers(self):

Instances