motif_sample.py

[msako82]

I downloaded the source code and Checkpoints files. I want to use the motif_sample.py, but it stops with the following error.

$ python motif_sample.py

[2024-03-24 01:11:33,984::sample::INFO] Namespace(config='./configs/sample.yml', data_id=1, device='cuda:0', num_workers=64, outdir='./outputs', vocab_path='vocab.txt')
[2024-03-24 01:11:33,984::sample::INFO] {'dataset': {'name': 'pl', 'path': './data/pdbbind_pocket10', 'split': './data/split_by_name.pt'}, 'model': {'checkpoint': './checkpoints/pretrained.pt', 'hidden_channels': 256, 'random_alpha': False}, 'sample': {'seed': 2024, 'num_samples': 100, 'num_retry': 5, 'max_steps': 12, 'batch_size': 10, 'num_workers': 4, 'n_samples': 5}}
[2024-03-24 01:11:33,984::sample::INFO] Loading data...
Segmentation fault (core dumped)

I checked the code in motif_sample.py and there is an error at line 507 where data = testset[args.data_id]

Do you have any ideas to solve this?