Pytorch version for loading data

SudeepDasari / RoboNet

MIT License

166 stars 20 forks source link

Pytorch version for loading data #7

Open lintangsutawika opened 4 years ago

lintangsutawika commented 4 years ago

Is there a plan to add a Pytorch version of the dataloading? If not, are you open to pull request?

SudeepDasari commented 4 years ago

I'm definitely interested in adding Pytorch and Tf-2 support (for dataloading/model training), but haven't gotten around to it yet. I'd certainly appreciate a pull request, if you have a solution in mind!