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