67eb9e4c28098d93ac122d65833c20b22b7e86c7,rl_coach/architectures/mxnet_components/architecture.py,MxnetArchitecture,_dummy_model_inputs,#MxnetArchitecture#,84

Before Change

        allowed_inputs["action"] = copy.copy(self.spaces.action)
        allowed_inputs["goal"] = copy.copy(self.spaces.goal)
        embedders = self.model.nets[0].input_embedders
        inputs = tuple(nd.zeros((1,) + tuple(allowed_inputs[emb.embedder_name].shape.tolist())) for emb in embedders)
        return inputs

    def construct_model(self) -> None:

After Change

        :return: tuple of inputs for model forward pass
        
        input_shapes = self._model_input_shapes()
        inputs = tuple(nd.zeros(tuple(shape)) for shape in input_shapes)
        return inputs

    def construct_model(self) -> None:

Instances