zhiyuanYU134 / MIRETR

19 stars 2 forks source link

getting Error: 'src_node_corr_knn_masks' #6

Open Rahul9564 opened 1 month ago

Rahul9564 commented 1 month ago

When running the code , getting the error Error: 'src_node_corr_knn_masks' in loss.py file : line 24

zhiyuanYU134 commented 1 month ago

When running the code , getting the error Error: 'src_node_corr_knn_masks' in loss.py file : line 24

Thank you for pointing out the error in the code. You can add the following code to line 402 of model.py, and I will fix the error in the corresponding file.: output_dict['src_node_corr_knn_masks']=src_neighbor_knn_masks