2020-01-06 06:59:35 +01: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-05-05 17:04:13 +02:00
|
|
|
// DisableTensorFlow returns true if TensorFlow should not be used for image classification.
|
2020-04-12 18:00:31 +02:00
|
|
|
func (c *Config) DisableTensorFlow() bool {
|
2020-04-13 18:08:21 +02:00
|
|
|
return c.params.DisableTensorFlow
|
2020-01-06 06:59:35 +01:00
|
|
|
}
|