Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

46 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

No description, website, or topics provided.

Resources

Stars

5 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages