utils/etc.py

import os
import torch
import time
import numpy as np
import json
def set_deterministic(seed):
    torch.manual_seed(seed)
    np.random.seed(seed)

def get_ssl_type(ssl_type):
    ssl_book = {
        "wavlm-large": "microsoft/wavlm-large",
        "wav2vec2-large": "facebook/wav2vec2-large-960h",
        "hubert": "facebook/hubert-large-ls960-ft"
    }
    return ssl_book.get(ssl_type, None)