1.Running Environments

  python 3.10.4

  pytorch 1.8.1

  NVIDIA A100 PCIe

  Ubuntu 18.04.6 LTS

2.Main Files

  main_stage1.py: The training file of mutual enhancement stage

  main_stage2.py: The training file of end-to-end aggregation stage

  pretrain.py: The pre-training file.

  models.py: Defining the transformer encoder.

  utils.py: Defining the methods.

  dataset.py: The data loading file.

3.Codes

  GdNeon/TRACE (github.com)