For the `torch_dist_full.py` how do you execute it, simply `python3 torch_dist_full.py` or are we supposed to use torch run etc.