File size: 849 Bytes
3ce12fe 9daa8bd 3ce12fe 2644561 9daa8bd |
1 2 3 4 5 6 7 8 9 10 11 |
The model consumes a lot of resources for inference, so its inference code was modified to split the input into smaller parts, which results in longer inference time. The model might not have been fully trained, so the quality may not be optimal. It is currently paused at checkpoint 8694, but further training may be conducted in the future. The djcm.pt model has been trimmed by removing to_wav since it is not used. Training Dataset: https://huggingface.co/datasets/AnhP/Mir-1k-use-DJCM-training/resolve/main/dataset.zip Inference Code: https://github.com/PhamHuynhAnh16/DJCM/blob/main/inference.py Training Code: https://github.com/PhamHuynhAnh16/DJCM/blob/main/train.py Export Small Code: https://github.com/PhamHuynhAnh16/DJCM/blob/main/export_small_model.py Export Onnx Code: https://github.com/PhamHuynhAnh16/DJCM/blob/main/export_onnx.py |