I am getting errors related to PyTorch-Geometric (PyG) version. I am using 2.0.2. m.weight does not work. It works when I change it to m.lin.weight, but the order in the model.modules() changes as well. So, all if k==2 parts need to be changed. It will be easier to use older version of PyG.
Are you using 1.6.1 as in the env.yml file? Or an updated version?