The platform we use is Python==3.9.18, PyTorch==2.2.2, and CUDA 12.0.
Running Instructions:
(1) Pre-trained models are located in the models/model/ folder.
(2) The dataset already includes some test data.
(3) To test the FRUFN model:
1) For MFIF task testing, run test_frufn_mfif.py.
2) For MEIF task testing, run test_frufn_meif.py.
3) For IVF task testing, run test_frufn_ivf.py.
(4) To train the FRUFN model:
1) First, train on the MFIF task by running train_frufn_mfif.py.
2) Then, train on the MEIF task by running train_frufn_meif_stage1.py.
3) Next, continue training on the MEIF task with train_frufn_meif_stage2.py.
4) Finally, train on the IVF task using train_frufn_ivf.py.
Testing and training instructions for the EFRUFN model are the same as those for the FRUFN model. Simply replace frufn with efrufn in the above test/train commands.