From c143c4bc0a7508e0dc88e13329dde9f1cc8a659b Mon Sep 17 00:00:00 2001 From: jyaacoub Date: Fri, 15 Dec 2023 12:40:40 -0500 Subject: [PATCH] style: print config info before run #68 --- rayTrain_Tune.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/rayTrain_Tune.py b/rayTrain_Tune.py index 6cde9af0..79b4e129 100644 --- 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__)