[FIX] bad argument error in dataset mapper

wbenbihi · Aug 20, 2022 · 68ab58c · 68ab58c
1 parent 60da17a
commit 68ab58c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hourglass_tensorflow/datasets/_meta.py b/hourglass_tensorflow/datasets/_meta.py
@@ -379,7 +379,7 @@ def _create_dataset(self, dataset: CastableTableDataset) -> tf.data.Dataset:
                     img,
                     coord,
                     vis,
-                    input_size=int(self.config.input_size),
+                    output_size=int(self.config.output_size),
                     stacks=self.config.stacks,
                     stddev=self.config.heatmap_stddev,
                 )