Skip to content
This repository has been archived by the owner on May 1, 2023. It is now read-only.

Load different random subset of dataset on each epoch (CIFAR-10 / ImageNet only) #149

Merged
merged 3 commits into from
Feb 10, 2019

Conversation

guyjacob
Copy link
Contributor

@guyjacob guyjacob commented Feb 7, 2019

No description provided.

* For CIFAR10 and ImageNet only
* Implemented custom sampler
* Integrated in image classification sample
@guyjacob guyjacob requested a review from nzmora February 7, 2019 12:40
Copy link
Contributor

@nzmora nzmora left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good - thanks!
Checkout my comment about float_range - I think the code that I offer there is a bit more generic.

examples/classifier_compression/parser.py Outdated Show resolved Hide resolved
* Changed randomization to match PyTorch's SubsetRandomSampler
* Generic float_range in parser
* Since we now shuffle the test set, had to update expected results
  in 2 full_flow_tests that do evaluation
@guyjacob guyjacob merged commit 4b1d0c8 into master Feb 10, 2019
@guyjacob guyjacob deleted the partial_datasets branch February 10, 2019 10:52
michaelbeale-IL pushed a commit that referenced this pull request Apr 24, 2023
* For CIFAR-10 / ImageNet only
* Refactor data_loaders.py, reduce code duplication
* Implemented custom sampler
* Integrated in image classification sample
* Since we now shuffle the test set, had to update expected results
  in 2 full_flow_tests that do evaluation
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants