Bug fix in collate().
Bug fix in collate().

Python was complaining when trying to run collate() because in line 511, one of the dimensions inside the reshape call (the division) was a float instead of an integer.

cespaillat/EDGE,cespaillat/EDGE