Method Save

model/model.go:966–1016

(dir string)

}

func (m *TfkgModel) Save(dir string) error {
	signatureCount := len(m.model.Signatures)

	e := os.MkdirAll(filepath.Join(dir, "variables"), os.ModePerm)
	if e != nil {
		m.errorHandler.Error(e)
		return e
	}

	if len(m.cpuPbCache) > 0 {
		e = ioutil.WriteFile(filepath.Join(dir, "saved_model.pb"), m.cpuPbCache, os.ModePerm)
		if e != nil {
			m.errorHandler.Error(e)
			return e
		}
	} else {
		e = ioutil.WriteFile(filepath.Join(dir, "saved_model.pb"), m.pbCache, os.ModePerm)
		if e != nil {
			m.errorHandler.Error(e)
			return e
		}
	}

	filenameInput, e := tf.NewTensor(filepath.Join(dir, "variables/variables"))
	if e != nil {
		m.errorHandler.Error(e)
		return e
	}

	_, e = m.model.Session.Run(
		map[tf.Output]*tf.Tensor{
			m.model.Graph.Operation("saver_filename").Output(0): filenameInput,
		},
		[]tf.Output{
			m.model.Graph.Operation(fmt.Sprintf("StatefulPartitionedCall_%d", signatureCount-1)).Output(0),
		},
		nil,
	)
	if e != nil {
		m.errorHandler.Error(e)
		return e
	}

	e = ioutil.WriteFile(filepath.Join(dir, "tfkg-version"), []byte(Version), os.ModePerm)
	if e != nil {
		m.errorHandler.Error(e)
		return e
	}

	return nil
}

type pythonConfig struct {
	ModelConfig            string      `json:"model_config"`