Merge branch 'correct_scripts' into 'VTM-11.0_nnvc'

Correct scripts

See merge request jvet-ahg-nnvc/VVCSoftware_VTM!115

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
 

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)