search

anniesch / jtt

Code for "Just Train Twice: Improving Group Robustness without Training Group Information"
67 stars 17 forks source link

Update CelebA dataloader to work with provided csvs #11

Open ieee8023 opened 6 months ago

ieee8023 commented 6 months ago

I added the code from this comment: https://github.com/anniesch/jtt/issues/1#issuecomment-1112269348

Also removed the numpy reference for in to work with the current version of numpy.