Add function to set logging level

nidn/__init__.py

@@ -1,8 +1,6 @@
 import os
-
-# Set main device by default to cpu if no other choice was made before
-if "TORCH_DEVICE" not in os.environ:
-    os.environ["TORCH_DEVICE"] = "cpu"
+import torch
+from loguru import logger
 
 # Add exposed features here
 from .plots.plot_model_grid import plot_model_grid
@@ -15,6 +13,19 @@
 from .utils.fix_random_seeds import fix_random_seeds
 from .utils.load_default_cfg import load_default_cfg
 from .utils.print_cfg import print_cfg
+from .utils.set_log_level import set_log_level
+
+set_log_level("INFO")
+
+# Set main device by default to cpu if no other choice was made before
+if "TORCH_DEVICE" not in os.environ:
+    os.environ["TORCH_DEVICE"] = "cpu"
+
+logger.info(f"Initialized NIDN for {os.environ['TORCH_DEVICE']}")
+
+# Set precision (and potentially GPU)
+torch.set_default_tensor_type(torch.DoubleTensor)
+logger.info("Using double precision")
 
 __all__ = [
     "compute_target_frequencies",
@@ -28,5 +39,6 @@
     "plot_model_grid_per_freq",
     "plot_spectra",
     "print_cfg",
+    "set_log_level",
     "wl_to_phys_wl",
 ]