'list' object has no attribute 'complex_t'

[lindafei01]

I encounter the following error message when running train.py Traceback (most recent call last): File "/root/autodl-tmp/DiffDock/train.py", line 161, in main_function() File "/root/autodl-tmp/DiffDock/train.py", line 156, in main_function train(args, model, optimizer, scheduler, ema_weights, train_loader, val_loader, t_to_sigma, run_dir) File "/root/autodl-tmp/DiffDock/train.py", line 35, in train train_losses = train_epoch(model, train_loader, optimizer, device, t_to_sigma, loss_fn, ema_weights) File "/root/autodl-tmp/DiffDock/utils/training.py", line 106, in train_epoch tr_pred, rot_pred, tor_pred = model(data) File "/root/miniconda3/envs/DiffDock/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(input, **kwargs) File "/root/autodl-tmp/DiffDock/models/score_model.py", line 236, in forward tr_sigma, rot_sigma, tor_sigma = self.t_to_sigma([data.complex_t[noise_type] for noise_type in ['tr', 'rot', 'tor']]) File "/root/autodl-tmp/DiffDock/models/score_model.py", line 236, in tr_sigma, rot_sigma, tor_sigma = self.t_to_sigma(*[data.complex_t[noise_type] for noise_type in ['tr', 'rot', 'tor']]) AttributeError: 'list' object has no attribute 'complex_t'