Method CompileAndLoad

model/model.go:1036–1202

(config CompileConfig, sessionOptions ...*for_core_protos_go_proto.ConfigProto)

}

func (m *TfkgModel) CompileAndLoad(config CompileConfig, sessionOptions ...*for_core_protos_go_proto.ConfigProto) error {
	if config.Loss == "" {
		config.Loss = LossMSE
	}
	if config.Optimizer == nil {
		config.Optimizer = optimizer.Adam()
	}
	if config.BatchSize == 0 {
		config.BatchSize = 1
	}
	m.logger.InfoF("model", "Compiling and loading model. If anything goes wrong python error messages will be printed out.")
	m.modelDefinitionSaveDir = config.ModelInfoSaveDir
	modelConfig, e := m.generateKerasDefinitionJson()
	if e != nil {
		return e
	}

	tempDir := filepath.Join(os.TempDir(), "/tfkg")

	e = os.MkdirAll(tempDir, os.ModePerm)
	if e != nil {
		m.errorHandler.Error(e)
		return e
	}

	if config.ModelInfoSaveDir != "" {
		indentedJson := bytes.NewBuffer([]byte{})
		e = json.Indent(indentedJson, []byte(modelConfig), "", "  ")
		if e != nil {
			m.errorHandler.Error(e)
			return e
		}
		e = ioutil.WriteFile(filepath.Join(config.ModelInfoSaveDir, "model.json"), indentedJson.Bytes(), os.ModePerm)
		if e != nil {
			m.errorHandler.Error(e)
			return e
		}
	}

	pConfig := pythonConfig{
		ModelConfig:            modelConfig,
		SaveDir:                filepath.Join(tempDir, tempModelDir),
		ModelDefinitionSaveDir: config.ModelInfoSaveDir,
		Loss:                   string(config.Loss),
		Optimizer:              config.Optimizer.GetKerasLayerConfig(),
		BatchSize:              config.BatchSize,
		CpuInference:           config.CpuInference,
	}

	configBytes, e := json.Marshal(pConfig)
	if e != nil {
		m.errorHandler.Error(e)
		return e
	}

	ignoreRegex := regexp.MustCompile("# tfkg-ignore.*# tfkg-ignore-end")

	layerTypesDefined := make(map[string]bool)