A small bugfix for the generate type

training/data_loader/data_loader.py

@@ -57,7 +57,7 @@ class DataLoader:
     # patch_size in luma sample
     def __init__(self, jsonfile, patch_size, poc_list, generate_type = 0, qp_filter=-1, slice_type_filter=-1):
         self.generate_type=generate_type
-        if self.generate_type:
+        if self.generate_type == 0:
             self.normalizer_rec  = 1023.0
             self.normalizer_pred = 1023.0
             self.normalizer_bs   = 1023.0
@@ -123,7 +123,7 @@ class DataLoader:
          self.database=dcontent['data']
          # create array of patches adress
         
-        if self.generate_type:
+        if self.generate_type == 0:
             psize = self.patch_size
             for didx in range(len(self.database)):
                  d=self.database[didx]