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,
)
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,
)