Fix my_data_loader to support newer versions of Pytorch

mit-han-lab / once-for-all

[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment

https://ofa.mit.edu/

MIT License

1.89k stars 333 forks source link

Closed jpablomch closed 4 years ago

jpablomch commented 4 years ago

This PR fixes data loader to distributed training using newer versions of Pytorch.