class_docstr: Evaluate losses on data pages
class_docstr_tok: ["Evaluate","losses","on","data","pages"]
code: def __init__(self, trainer: MDNRNNTrainer, state_dim: int) -> None:
      self.trainer = trainer
      self.state_dim = state_dim
  
file_path: ReAgent/reagent/evaluation/world_model_evaluator.py
imports: import logging
  from typing import Dict, List
  import torch
  from reagent.core.types import FeatureData, MemoryNetworkInput
  from reagent.training.world_model.mdnrnn_trainer import MDNRNNTrainer
label_desc: Use image node_id 1 to create a new LossEvaluator object  with example: obj = LossEvaluator(trainer, state_dim)