Get errors when training the model: AttributeError: Can't get attribute 'LoadSource' on <module 'trimesh.parent'

[yangrongkun]

Loading data from logs/4ac080a4f/checkpoints/last.ckpt
Traceback (most recent call last):
File "optimize_ckpt.py", line 140, in
main(args)
File "optimize_ckpt.py", line 113, in main
fit_ckpt(args)
File "optimize_ckpt.py", line 12, in fit_ckpt
out, ckpt = load_data(args.ckpt_p)
File "/data2/program/hold/code/src/utils/io/optim.py", line 27, in load_data
misc = np.load(misc_ps[-1], allow_pickle=True).item()
File "/home/yrk/anaconda3/envs/hold/lib/python3.8/site-packages/numpy/lib/npyio.py", line 432, in load
return format.read_array(fid, allow_pickle=allow_pickle,
File "/home/yrk/anaconda3/envs/hold/lib/python3.8/site-packages/numpy/lib/format.py", line 781, in read_array
array = pickle.load(fp, **pickle_kwargs)
AttributeError: Can't get attribute 'LoadSource' on <module 'trimesh.parent' from '/home/yrk/anaconda3/envs/hold/lib/python3.8/site-packages/trimesh/parent.py'>

[Peraspera1]

I encountered the same problem and solved it by inserting the following code at the beginning of visualize_ckpt.py:

try:
    import trimesh.parent
    class LoadSource:
        def __init__(self, *args, **kwargs):
            pass
    trimesh.parent.LoadSource = LoadSource
except (ImportError, AttributeError):
    pass

Hope it's useful to you!

[yangrongkun]

Thank you very much!!! This works for me.