From 8557c562574c6f03f1eef227fec3821eb351e9f3 Mon Sep 17 00:00:00 2001
From: Franck Galpin <franck.galpin@interdigital.com>
Date: Thu, 15 Jun 2023 08:55:07 +0000
Subject: [PATCH] Correct scripts
---
training/tools/dataset_to_encoding_script.py | 6 +++++-
.../training_scripts/NN_Filtering_HOP/training/trainer.py | 2 +-
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/training/tools/dataset_to_encoding_script.py b/training/tools/dataset_to_encoding_script.py
index 59a99ab079..042a95b01c 100644
--- a/training/tools/dataset_to_encoding_script.py
+++ b/training/tools/dataset_to_encoding_script.py
@@ -212,10 +212,14 @@ for seq in dataset:
echo "[CACHED] {basename}_QP{qp} already encoded"
fi
if [ ! -f ${{DECDIR}}/{basename}_QP{qp}.log ]; then
+ if [ -f ${{ENCDIR}}/{basename}_QP{qp}.bin -a ${{ENCDIR}}/{basename}_QP{qp}.log ]; then
$DEC -b ${{ENCDIR}}/{basename}_QP{qp}.bin --DumpBasename=${{DUMPDIR}}/{basename}_QP{qp} $OPTDEC > ${{DECDIR}}/{basename}_QP{qp}.log
echo "[INFO] {basename}_QP{qp} decoded"
else
- echo "[CACHED] {basename}_QP{qp} already decoded"
+ echo "[INFO] skipped decoding (no bitstream)";
+ fi;
+ else
+ echo "[CACHED] {basename}_QP{qp} already decoded";
fi
fi
diff --git a/training/training_scripts/NN_Filtering_HOP/training/trainer.py b/training/training_scripts/NN_Filtering_HOP/training/trainer.py
index 3834ab27d2..d92c91ccfe 100644
--- a/training/training_scripts/NN_Filtering_HOP/training/trainer.py
+++ b/training/training_scripts/NN_Filtering_HOP/training/trainer.py
@@ -65,7 +65,7 @@ class Trainer:
self.device = self.config_training["device"] or (
torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
)
-
+ print(f"[INFO] tf32 {torch.backends.cuda.matmul.allow_tf32} {torch.backends.cudnn.allow_tf32}")
self.base_dir = self.config_training["path"]
self.save_dir = os.path.join(self.base_dir, self.config_training["ckpt_dir"])
os.makedirs(self.save_dir, exist_ok=True)
--
GitLab