model load error (need to call init_process_group? to load saved model with DDP lightning)

[yusuke0324]

Hi

After this traini $python src/main.py +experiment=ego.yaml general.gps=[0,1,2],

I tried to load visualize using the following command,

$python main.py +experiment=ego.yaml general.sample_visualization=True general.test_only=/work/outputs/comm20/2025-06-01/10-49-11-graph-tf-model/checkpoints/graph-tf-model/epoch\=99-v1.ckpt

I got the error:

RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.

I think this is related to the issue from Digress.
Do you know how to deal with the issue?
Pytorch version is '2.0.1+cu118' and Pytorch lightning version is '2.0.4'

Thanks!