search

vikasverma1077 / manifold_mixup

Code for reproducing Manifold Mixup results (ICML 2019)
483 stars 65 forks source link

A bunch of small updates #17

Open NetradeepakC opened 7 months ago

NetradeepakC commented 7 months ago
  1. Changed deprecated code to make code compatible with Pytorch 2
  2. Created conda environment file which will guarantee the model functioning in the future
  3. Created .gitignore to keep datasets and experiments outside of GitHub
  4. Added code to arg parser to select layers for more intricate experiments
  5. Re-enabled option for data parallelism
  6. Designed an improved scheduler to produce results better than the research paper
  7. Created more thorough testing of alpha values for mixups to select the best alpha