style: print config info before run #68

jyaacoub · Dec 15, 2023 · c143c4b · c143c4b
1 parent ecf586c
commit c143c4b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rayTrain_Tune.py b/rayTrain_Tune.py
@@ -66,7 +66,9 @@ def train_func(config):
         ray.train.report({"loss": loss},   checkpoint=checkpoint)
 
 
-if __name__ == "__main__":    
+if __name__ == "__main__":
+    print("DATA_ROOT:", cfg.DATA_ROOT)
+    print("os.environ['TRANSFORMERS_CACHE']", os.environ['TRANSFORMERS_CACHE'])
     print("Cuda support:", torch.cuda.is_available(),":", 
                             torch.cuda.device_count(), "devices")
     print("CUDA VERSION:", torch.__version__)