* Fix bug in Collector with Graph data * Add comments in DataModule class and bug fix in collate
* Add num_workers and pin_memory arguments to DataLoader and DataModule tests