Prepare for hpc test

tu-studio · Jul 29, 2024 · 35b581a · 35b581a
1 parent d8dcb2e
commit 35b581a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 8 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,16 +3,15 @@
 __pycache__/
 
 # Ignore Python virtual environments
-myenv
 venv
 
 # Ignore data files
 data/*/*
 logs/*/*
 /dvclive
 /models
-/temp
-/exp-logs/*
+/tmp
+/exp_logs/*
 
 # Not ignore
 !logs/*/.gitkeep

diff --git a/TODO.md b/TODO.md
diff --git a/dvc.yaml b/dvc.yaml
@@ -39,5 +39,5 @@ stages:
   save_logs:
     cmd: python3 source/utils/logs.py
     outs:
-    - exp_logs/slurm
+#    - exp_logs/slurm
     - exp_logs/tensorboard/
diff --git a/source/utils/logs.py b/source/utils/logs.py
@@ -3,7 +3,7 @@
 import torch
 from torch.utils.tensorboard import SummaryWriter
 from torch.utils.tensorboard.summary import hparams
-from utils.config import get_env_variable
+from config import get_env_variable
 
 
 # Overrides the Tensorboard SummaryWriter class to add hyperparameters to the same tensorboard logs and enable metrics as scalar sequences