correction for hop5

training/training_scripts/NN_Filtering/common/training/trainer.py

@@ -325,8 +325,8 @@ class Trainer:
         return {"lossY": lossY, "lossUV": lossUV, "lossYUV": lossYUV}
 
     def iteration(self, sample):
+        cropped_size = False
         if "dct_size" in self.config_training and self.dct_size >= 2:
-            cropped_size = False
             sample_dct = self.applyDCT2(sample)
             Ycoeff_res, UVcoeff_res = self.model(
                 {name: tensor.to(self.device) for name, tensor in sample_dct.items()}