Full Training Example

Concise steps to train the model. Make sure the environment is installed.

An example training:

cd /home/SIT_FUSE/sit_fuse/datasets
python3 sf_dataset.py -y ../config/model/gk2a_test_dbn_multi_layer_pl.yaml
cd ../train
python3 pretrain_encoder.py -y ../config/model/gk2a_test_dbn_multi_layer_pl.yaml
python3 finetune_dc_mlp_head.py -y ../config/model/gk2a_test_dbn_multi_layer_pl.yaml
python3 train_heirarchichal_deep_cluster.py -y ../config/model/gk2a_test_dbn_multi_layer_pl.yaml
cd ../inference
python3 generate_output.py -y ../config/model/gk2a_test_dbn_multi_layer_pl.yaml

Training Dataset PreprocessingTraining CommandInference Command