2020-01-06 05:59:35 +00:00
|
|
|
package config
|
|
|
|
|
|
|
|
import tf "github.com/tensorflow/tensorflow/tensorflow/go"
|
|
|
|
|
|
|
|
// TensorFlowVersion returns the TenorFlow framework version.
|
|
|
|
func (c *Config) TensorFlowVersion() string {
|
|
|
|
return tf.Version()
|
|
|
|
}
|
|
|
|
|
2020-04-12 16:00:31 +00:00
|
|
|
// DisableTensorFlow returns true if the use of TensorFlow is disabled for image classification.
|
|
|
|
func (c *Config) DisableTensorFlow() bool {
|
2020-01-06 05:59:35 +00:00
|
|
|
return c.config.DisableTensorFlow
|
|
|
|
}
|