* Update torch_utils.py * Additional code refactoring * tuples to sets * Cleanup
utils/datasets.py
utils/dataloaders.py