pickle.UnpicklingError: invalid load key, 'v'

Hi,

Thanks for sharing such an interesting repo, I tried to test the approach for my data, and ran into the following invalid code when reading some DCT stats (?)

Any idea what is going on?

python experiments/correct_batch.py --input $INPUT_DIR test --output $OUTPUT_DIR --weights ./weights/weights_c.pt --dstats_path ./stats/cstats.pt --color
Traceback (most recent call last):
  File "experiments/correct_batch.py", line 51, in <module>
    stats = dct.DCTStats(args.dstats_path, type="ms")
  File "quantization-guided-ac/dct/dct_stats.py", line 7, in __init__
    super(DCTStats, self).__init__(root, type)
  File "quantization-guided-ac/dct/stats.py", line 7, in __init__
    self.stats = torch.load(root)
  File "/home/petteri/anaconda3/lib/python3.8/site-packages/torch/serialization.py", line 593, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/home/petteri/anaconda3/lib/python3.8/site-packages/torch/serialization.py", line 762, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.