2c7a2997c28d0be732e4157e924dce87bcf56236,ml/rl/training/ml_trainer.py,MLTrainer,build_predictor,#MLTrainer#Any#Any#Any#,207

Before Change

                workspace.FeedBlob(weight_name, weights)

    def build_predictor(self, model, input_blob, output_blob) -> List[str]:
        MakeForwardPassOps(
            model,
            "predictor",
            input_blob,
            output_blob,
            self.weights,
            self.biases,
            self.activations,
            self.layers,
            self.dropout_ratio,
            is_test=True
        )
        return self.weights + self.biases

After Change

            build_sgd(model, self.learning_rate)

    def build_predictor(self, model, input_blob, output_blob) -> List[str]:
        self.make_forward_pass_ops(model, input_blob, output_blob, is_test=True)
        return self.weights + self.biases

Instances