environment:
python 3.9

folders:
1) checkpoints/: pre-trained weights
2) config/: configs for model & training
3) data/: datasets
4) flow/:
flow/models: SiT models
flow/transport: modules for flow-based models
5) layers/: layers for conditional featue extractor
6) ldns: simulated data generation
7) model/: conditional feature extractor
8) utils/: masking schemes for conditional featue extractor

scripts:
1) simulated neural data:
pre-train: main_pretrain_synthetic.py
fine-tuning: main_ft_synthetic.py
2) NLB21: main_pretrain_nlb21.py
3) Falcon:
pre-train: main_pretrain_falcon.py
fine-tuning: main_ft_falcon.py