├── .gitignore ├── LICENSE ├── bins ├── infer.py └── train.py ├── configs ├── ds_stage2.json ├── sac_16k_37_5.yaml └── sac_16k_62_5.yaml ├── docs ├── batch_scripts_guide.md ├── evaluation_guide.md └── feature_extraction_guide.md ├── example ├── audio-en.wav ├── audio-en_recon.wav ├── audio-zh.wav ├── audio-zh_recon.wav ├── batch_script_data.jsonl └── training_data.jsonl ├── models ├── base │ ├── __init__.py │ ├── base_dataloader.py │ ├── base_datapipes.py │ ├── base_trainer.py │ ├── fsq │ │ ├── finite_scalar_quantization.py │ │ └── residual_fsq.py │ ├── loss │ │ ├── __init__.py │ │ ├── reconstruct.py │ │ └── ssim.py │ └── modules │ │ ├── dac.py │ │ ├── dac_utils │ │ ├── __init__.py │ │ ├── base.py │ │ ├── layers.py │ │ ├── loss.py │ │ └── quantize.py │ │ ├── ecapa_tdnn.py │ │ ├── fsq_encoder.py │ │ ├── perceiver_encoder.py │ │ └── pooling_layers.py └── codec │ ├── base │ ├── base_codec_trainer.py │ └── quantizer │ │ ├── __init__.py │ │ ├── distrib.py │ │ ├── factorized_vector_quantize.py │ │ └── quantizer_gumbel.py │ └── sac │ ├── blocks │ ├── layers.py │ └── loss.py │ ├── dataloader.py │ ├── eval │ ├── __init__.py │ └── metrics.py │ ├── model.py │ ├── modules │ ├── acoustic_encoder.py │ ├── decoder.py │ ├── encoder.py │ ├── sampler.py │ ├── semantic_encoder.py │ ├── speaker_encoder.py │ ├── speaker_predictor.py │ ├── utils │ │ ├── ERes2Net.py │ │ ├── fbank.py │ │ ├── fusion.py │ │ └── pooling_layers.py │ ├── vocoder │ │ ├── msstft_discriminator.py │ │ ├── stft_utils.py │ │ ├── vocos_decoder.py │ │ ├── wave_discriminator.py │ │ └── wave_generator.py │ └── vocos.py │ ├── pipelines │ ├── decode_tokens.py │ ├── encode_tokens.py │ ├── extract_embeddings.py │ └── reconstruct.py │ ├── third_party │ └── hf_whisper │ │ ├── __init__.py │ │ ├── configuration_whisper.py │ │ ├── generation_whisper.py │ │ └── modeling_whisper.py │ ├── trainer.py │ └── utils.py ├── readme.md ├── requirements.txt ├── scripts ├── batch │ ├── decode.sh │ ├── encode.sh │ ├── extract_embeddings.sh │ └── reconstruct.sh ├── eval.sh └── train.sh ├── tools ├── evaluation │ ├── base_evaluator.py │ ├── metircs │ │ ├── compute_wer.py │ │ ├── periodicity.py │ │ └── utmos.py │ └── speech_evaluator.py ├── features │ ├── extract_tokens.py │ ├── glm-4-voice_feat.py │ └── modeling_whisper.py └── speaker │ ├── extract_spk_emb.py │ ├── spk_sim_cal.py │ └── utils │ ├── __init__.py │ ├── ecapa_tdnn.py │ └── utils.py └── utils ├── __init__.py ├── audio.py ├── checkpoint.py ├── commons.py ├── data_processor.py ├── file.py ├── log.py ├── parse_options.sh ├── plot.py ├── scheduler.py └── train_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/LICENSE -------------------------------------------------------------------------------- /bins/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/bins/infer.py -------------------------------------------------------------------------------- /bins/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/bins/train.py -------------------------------------------------------------------------------- /configs/ds_stage2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/configs/ds_stage2.json -------------------------------------------------------------------------------- /configs/sac_16k_37_5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/configs/sac_16k_37_5.yaml -------------------------------------------------------------------------------- /configs/sac_16k_62_5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/configs/sac_16k_62_5.yaml -------------------------------------------------------------------------------- /docs/batch_scripts_guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/docs/batch_scripts_guide.md -------------------------------------------------------------------------------- /docs/evaluation_guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/docs/evaluation_guide.md -------------------------------------------------------------------------------- /docs/feature_extraction_guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/docs/feature_extraction_guide.md -------------------------------------------------------------------------------- /example/audio-en.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/example/audio-en.wav -------------------------------------------------------------------------------- /example/audio-en_recon.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/example/audio-en_recon.wav -------------------------------------------------------------------------------- /example/audio-zh.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/example/audio-zh.wav -------------------------------------------------------------------------------- /example/audio-zh_recon.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/example/audio-zh_recon.wav -------------------------------------------------------------------------------- /example/batch_script_data.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/example/batch_script_data.jsonl -------------------------------------------------------------------------------- /example/training_data.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/example/training_data.jsonl -------------------------------------------------------------------------------- /models/base/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base/base_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/base_dataloader.py -------------------------------------------------------------------------------- /models/base/base_datapipes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/base_datapipes.py -------------------------------------------------------------------------------- /models/base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/base_trainer.py -------------------------------------------------------------------------------- /models/base/fsq/finite_scalar_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/fsq/finite_scalar_quantization.py -------------------------------------------------------------------------------- /models/base/fsq/residual_fsq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/fsq/residual_fsq.py -------------------------------------------------------------------------------- /models/base/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base/loss/reconstruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/loss/reconstruct.py -------------------------------------------------------------------------------- /models/base/loss/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/loss/ssim.py -------------------------------------------------------------------------------- /models/base/modules/dac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/dac.py -------------------------------------------------------------------------------- /models/base/modules/dac_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/dac_utils/__init__.py -------------------------------------------------------------------------------- /models/base/modules/dac_utils/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/dac_utils/base.py -------------------------------------------------------------------------------- /models/base/modules/dac_utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/dac_utils/layers.py -------------------------------------------------------------------------------- /models/base/modules/dac_utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/dac_utils/loss.py -------------------------------------------------------------------------------- /models/base/modules/dac_utils/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/dac_utils/quantize.py -------------------------------------------------------------------------------- /models/base/modules/ecapa_tdnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/ecapa_tdnn.py -------------------------------------------------------------------------------- /models/base/modules/fsq_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/fsq_encoder.py -------------------------------------------------------------------------------- /models/base/modules/perceiver_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/perceiver_encoder.py -------------------------------------------------------------------------------- /models/base/modules/pooling_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/base/modules/pooling_layers.py -------------------------------------------------------------------------------- /models/codec/base/base_codec_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/base/base_codec_trainer.py -------------------------------------------------------------------------------- /models/codec/base/quantizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/codec/base/quantizer/distrib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/base/quantizer/distrib.py -------------------------------------------------------------------------------- /models/codec/base/quantizer/factorized_vector_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/base/quantizer/factorized_vector_quantize.py -------------------------------------------------------------------------------- /models/codec/base/quantizer/quantizer_gumbel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/base/quantizer/quantizer_gumbel.py -------------------------------------------------------------------------------- /models/codec/sac/blocks/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/blocks/layers.py -------------------------------------------------------------------------------- /models/codec/sac/blocks/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/blocks/loss.py -------------------------------------------------------------------------------- /models/codec/sac/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/dataloader.py -------------------------------------------------------------------------------- /models/codec/sac/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/codec/sac/eval/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/eval/metrics.py -------------------------------------------------------------------------------- /models/codec/sac/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/model.py -------------------------------------------------------------------------------- /models/codec/sac/modules/acoustic_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/acoustic_encoder.py -------------------------------------------------------------------------------- /models/codec/sac/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/decoder.py -------------------------------------------------------------------------------- /models/codec/sac/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/encoder.py -------------------------------------------------------------------------------- /models/codec/sac/modules/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/sampler.py -------------------------------------------------------------------------------- /models/codec/sac/modules/semantic_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/semantic_encoder.py -------------------------------------------------------------------------------- /models/codec/sac/modules/speaker_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/speaker_encoder.py -------------------------------------------------------------------------------- /models/codec/sac/modules/speaker_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/speaker_predictor.py -------------------------------------------------------------------------------- /models/codec/sac/modules/utils/ERes2Net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/utils/ERes2Net.py -------------------------------------------------------------------------------- /models/codec/sac/modules/utils/fbank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/utils/fbank.py -------------------------------------------------------------------------------- /models/codec/sac/modules/utils/fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/utils/fusion.py -------------------------------------------------------------------------------- /models/codec/sac/modules/utils/pooling_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/utils/pooling_layers.py -------------------------------------------------------------------------------- /models/codec/sac/modules/vocoder/msstft_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/vocoder/msstft_discriminator.py -------------------------------------------------------------------------------- /models/codec/sac/modules/vocoder/stft_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/vocoder/stft_utils.py -------------------------------------------------------------------------------- /models/codec/sac/modules/vocoder/vocos_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/vocoder/vocos_decoder.py -------------------------------------------------------------------------------- /models/codec/sac/modules/vocoder/wave_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/vocoder/wave_discriminator.py -------------------------------------------------------------------------------- /models/codec/sac/modules/vocoder/wave_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/vocoder/wave_generator.py -------------------------------------------------------------------------------- /models/codec/sac/modules/vocos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/modules/vocos.py -------------------------------------------------------------------------------- /models/codec/sac/pipelines/decode_tokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/pipelines/decode_tokens.py -------------------------------------------------------------------------------- /models/codec/sac/pipelines/encode_tokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/pipelines/encode_tokens.py -------------------------------------------------------------------------------- /models/codec/sac/pipelines/extract_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/pipelines/extract_embeddings.py -------------------------------------------------------------------------------- /models/codec/sac/pipelines/reconstruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/pipelines/reconstruct.py -------------------------------------------------------------------------------- /models/codec/sac/third_party/hf_whisper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/codec/sac/third_party/hf_whisper/configuration_whisper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/third_party/hf_whisper/configuration_whisper.py -------------------------------------------------------------------------------- /models/codec/sac/third_party/hf_whisper/generation_whisper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/third_party/hf_whisper/generation_whisper.py -------------------------------------------------------------------------------- /models/codec/sac/third_party/hf_whisper/modeling_whisper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/third_party/hf_whisper/modeling_whisper.py -------------------------------------------------------------------------------- /models/codec/sac/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/trainer.py -------------------------------------------------------------------------------- /models/codec/sac/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/models/codec/sac/utils.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/batch/decode.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/scripts/batch/decode.sh -------------------------------------------------------------------------------- /scripts/batch/encode.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/scripts/batch/encode.sh -------------------------------------------------------------------------------- /scripts/batch/extract_embeddings.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/scripts/batch/extract_embeddings.sh -------------------------------------------------------------------------------- /scripts/batch/reconstruct.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/scripts/batch/reconstruct.sh -------------------------------------------------------------------------------- /scripts/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/scripts/eval.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /tools/evaluation/base_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/evaluation/base_evaluator.py -------------------------------------------------------------------------------- /tools/evaluation/metircs/compute_wer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/evaluation/metircs/compute_wer.py -------------------------------------------------------------------------------- /tools/evaluation/metircs/periodicity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/evaluation/metircs/periodicity.py -------------------------------------------------------------------------------- /tools/evaluation/metircs/utmos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/evaluation/metircs/utmos.py -------------------------------------------------------------------------------- /tools/evaluation/speech_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/evaluation/speech_evaluator.py -------------------------------------------------------------------------------- /tools/features/extract_tokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/features/extract_tokens.py -------------------------------------------------------------------------------- /tools/features/glm-4-voice_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/features/glm-4-voice_feat.py -------------------------------------------------------------------------------- /tools/features/modeling_whisper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/features/modeling_whisper.py -------------------------------------------------------------------------------- /tools/speaker/extract_spk_emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/speaker/extract_spk_emb.py -------------------------------------------------------------------------------- /tools/speaker/spk_sim_cal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/speaker/spk_sim_cal.py -------------------------------------------------------------------------------- /tools/speaker/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/speaker/utils/ecapa_tdnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/speaker/utils/ecapa_tdnn.py -------------------------------------------------------------------------------- /tools/speaker/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/tools/speaker/utils/utils.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/audio.py -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/commons.py -------------------------------------------------------------------------------- /utils/data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/data_processor.py -------------------------------------------------------------------------------- /utils/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/file.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/parse_options.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/parse_options.sh -------------------------------------------------------------------------------- /utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/plot.py -------------------------------------------------------------------------------- /utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/scheduler.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Soul-AILab/SAC/HEAD/utils/train_utils.py --------------------------------------------------------------------------------