Exemplo n.º 1
0
 def build_batch_preprocessor(self) -> BatchPreprocessor:
     return PolicyNetworkBatchPreprocessor(
         state_preprocessor=Preprocessor(
             self.state_normalization_parameters, use_gpu=self.use_gpu),
         action_preprocessor=Preprocessor(
             self.action_normalization_parameters, use_gpu=self.use_gpu),
         use_gpu=self.use_gpu,
     )
Exemplo n.º 2
0
 def build_batch_preprocessor(self) -> BatchPreprocessor:
     state_preprocessor = Preprocessor(
         self.state_normalization_data.dense_normalization_parameters, )
     action_preprocessor = Preprocessor(
         self.action_normalization_data.dense_normalization_parameters, )
     return PolicyNetworkBatchPreprocessor(
         state_preprocessor=state_preprocessor,
         action_preprocessor=action_preprocessor,
     )