-
- Downloads
Load different random subset of dataset on each epoch (#149)
* 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
Showing
- apputils/data_loaders.py 104 additions, 91 deletionsapputils/data_loaders.py
- examples/agp-pruning/resnet20_filters.schedule_agp.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet20_filters.schedule_agp.yaml
- examples/agp-pruning/resnet20_filters.schedule_agp_3.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet20_filters.schedule_agp_3.yaml
- examples/agp-pruning/resnet20_filters.schedule_agp_4.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet20_filters.schedule_agp_4.yaml
- examples/agp-pruning/resnet50.schedule_agp.1x1x8-blocks.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet50.schedule_agp.1x1x8-blocks.yaml
- examples/agp-pruning/resnet50.schedule_agp.filters.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet50.schedule_agp.filters.yaml
- examples/agp-pruning/resnet50.schedule_agp.filters_2.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet50.schedule_agp.filters_2.yaml
- examples/agp-pruning/resnet50.schedule_agp.filters_3.yaml 1 addition, 1 deletionexamples/agp-pruning/resnet50.schedule_agp.filters_3.yaml
- examples/agp-pruning/resnet50.schedule_agp.filters_with_FC.yaml 1 addition, 1 deletion...es/agp-pruning/resnet50.schedule_agp.filters_with_FC.yaml
- examples/classifier_compression/compress_classifier.py 4 additions, 2 deletionsexamples/classifier_compression/compress_classifier.py
- examples/classifier_compression/parser.py 25 additions, 6 deletionsexamples/classifier_compression/parser.py
- examples/network_surgery/resnet20.network_surgery.yaml 1 addition, 1 deletionexamples/network_surgery/resnet20.network_surgery.yaml
- examples/network_surgery/resnet50.network_surgery.yaml 1 addition, 1 deletionexamples/network_surgery/resnet50.network_surgery.yaml
- examples/network_surgery/resnet50.network_surgery2.yaml 1 addition, 1 deletionexamples/network_surgery/resnet50.network_surgery2.yaml
- examples/network_trimming/resnet50.filters.activation_apoz_agp.yaml 1 addition, 1 deletion...etwork_trimming/resnet50.filters.activation_apoz_agp.yaml
- tests/full_flow_tests.py 2 additions, 2 deletionstests/full_flow_tests.py
Loading
Please register or sign in to comment