change to reflect new dataloader usage
change to reflect new dataloader usage

NervanaSystems/neon,Jokeren/neon,NervanaSystems/neon,Jokeren/neon,Jokeren/neon,NervanaSystems/neon,Jokeren/neon,NervanaSystems/neon