hm

hydra_node/config.py

@@ -17,6 +17,11 @@ def init_config_model():
     config.dtype = os.getenv("DTYPE", "float16")
     config.device = os.getenv("DEVICE", "cuda")
     config.amp = os.getenv("AMP", False)
+    if config.amp == "1":
+        config.amp = True
+    elif config.amp == "0":
+        config.amp = False
+        
     is_dev = ""
     environment = "production"
     if os.environ['DEV'] == "True":