KeyError: 'survival_bin'

[rongyua]

The following problem occurs when running CUDA_VISIBLE_DEVICES=2,3 nohup 2_HistoPath_train.py --config "/path/to/config_ffpe_train.json"

Traceback (most recent call last):
File "2_HistoPath_train.py", line 610, in
main()
File "2_HistoPath_train.py", line 591, in main
target_label=config.get('target_label', 'vital_status'),
File "2_HistoPath_train.py", line 313, in train_model
survival_bin = batch_dict['survival_bin'].to(device).long()
KeyError: 'survival_bin'

Can you tell me what's going on, I'd be grateful if you could