class_docstr: Trainer for MDN-RNN
class_docstr_tok: ["Trainer","for","MDN-RNN"]
code: def __init__(
      self,
      memory_network: MemoryNetwork,
      params: MDNRNNTrainerParameters,
      cum_loss_hist: int = 100,
  ):
      super().__init__()
      self.memory_network = memory_network
      self.params = params
  
file_path: ReAgent/reagent/training/world_model/mdnrnn_trainer.py
imports: import logging
  from typing import Optional
  import reagent.core.types
  import torch
  import torch.nn.functional
  from reagent.core.parameters import MDNRNNTrainerParameters
  from reagent.models.mdn_rnn import gmm_loss
  from reagent.models.world_model import MemoryNetwork
  from reagent.training.reagent_lightning_module import ReAgentLightningModule
label_desc: Use image node_id 1 to create a new MDNRNNTrainer object from inherited base classes: ReAgentLightningModule with example: obj = MDNRNNTrainer(memory_network, params, cum_loss_hist)