Chia v2 pure

chillax/experiment_selfsupervised.py

@@ -1,6 +1,6 @@
 from chia.v2 import containers, instrumentation
 from chia.v2.components import classifiers
-from chia.framework.startup import tensorflow_global_settings
+from chia.v2 import helpers
 from chillax import method
 
 
@@ -42,7 +42,7 @@ def main():
     )
 
     # This shouldn't be necessary, but...
-    tensorflow_global_settings.startup_fn()
+    helpers.setup_environment()
 
     experiment_container = containers.ExperimentContainer(
         config, observers=(CheapObserver(),)