14 lines
386 B
Go
14 lines
386 B
Go
package config
|
|
|
|
import tf "github.com/tensorflow/tensorflow/tensorflow/go"
|
|
|
|
// TensorFlowVersion returns the TenorFlow framework version.
|
|
func (c *Config) TensorFlowVersion() string {
|
|
return tf.Version()
|
|
}
|
|
|
|
// DisableTensorFlow returns true if the use of TensorFlow is disabled for image classification.
|
|
func (c *Config) DisableTensorFlow() bool {
|
|
return c.params.DisableTensorFlow
|
|
}
|