fix multiprocessing test

fix multiprocessing test to accelerate test speed

second release to get better performer

This second version to fix some bugs like heatmaps.

first commit

release a pytorch version