Bar

This patch tries to fix a issue #176, among others. Refactored add_policy, and added a few other fixes that I use that patch with. Some insight: Originally this PR was...

Quantization stats collector fails if input is not a torch.Tensor

5

Failure occurs during qe_calibration of model Resnext101_32x4d, when called on classifier.py. update_record() expects a tensor, but gets a list of tensors instead. I have solved this by stacking up the...

Fails to resume training CPU-->GPU

2

When I try to resume training (with compress_classifier.py), load_checkpoint() fails to load the compress_scheduler, due to KeyError. This happens only when trying to resume training on GPU, when the checkpoint...