train running dataset fixed

2020-03-05 20:50:07 +01:00
parent 1f25bf599b
commit 05033bed75
11 changed files with 41 additions and 49 deletions
--- a/datasets/trajectory_dataset.py
+++ b/datasets/trajectory_dataset.py
@ -87,7 +87,7 @@ class TrajData(object):
        max_map_size = (1, ) + tuple(reversed(tuple(map(max, *[Image.open(map_file).size for map_file in map_files]))))
        return ConcatDataset([TrajDataset(maps_root=self.maps_root, mapname=map_file.name, length=equal_split,
                                          all_in_map=self.all_in_map, embedding_size=max_map_size,
-                                          preserve_equal_samples=True)
+                                          preserve_equal_samples=False)
                              for map_file in map_files])

    @property