Spaces:
Sleeping
Sleeping
changing cpu env
Browse files- ndvi_predictor.py +1 -1
ndvi_predictor.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# ndvi_predictor.py
|
| 2 |
import os
|
| 3 |
-
|
| 4 |
os.environ["SM_FRAMEWORK"] = "tf.keras"
|
| 5 |
import segmentation_models as sm
|
| 6 |
import tensorflow as tf
|
|
|
|
| 1 |
# ndvi_predictor.py
|
| 2 |
import os
|
| 3 |
+
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
|
| 4 |
os.environ["SM_FRAMEWORK"] = "tf.keras"
|
| 5 |
import segmentation_models as sm
|
| 6 |
import tensorflow as tf
|