jaeyeonkim99 / EnCLAP

Official Implementation of EnCLAP (ICASSP 2024)
MIT License
88 stars 5 forks source link

Hello, what does the args'--ckpt' mean in the inference.py "python inference.py --ckpt checkpoint_directory_with_config --clap_ckpt clap_ckpt_path --input input.wav"? #4

Closed DthdZK closed 5 months ago

jaeyeonkim99 commented 6 months ago

Sorry for the late reply.

"--ckpt" refers to the checkpoint folder (e.g. "output/checkpoints/epoch_5") containing both the weight file (i.e., "pytorch_model.bin") and the config file (i.e., "config.json") of the model. Both files are automatically saved during training inside the output directory.

You can also refer to pretrained checkpoints as examples.

Thank you.