postgres to torch dataset tests and script for billion-words dataset