Skip to content

Instantly share code, notes, and snippets.

View eschmidbauer's full-sized avatar

Emmanuel Schmidbauer eschmidbauer

View GitHub Profile
from collections import OrderedDict
import torch
checkpoint_path = "multilingual_base_bs100x4.ckpt"
checkpoint = torch.load(checkpoint_path, map_location=torch.device('cpu'))
modified_checkpoint = OrderedDict()
modified_checkpoint['state_dict'] = OrderedDict()
state_dict = checkpoint['state_dict'].copy()
for k, v in state_dict.items():
k = f'net.{k}'
import multiprocessing
from concurrent.futures import ProcessPoolExecutor, as_completed
# !pip install -q torchaudio
SAMPLING_RATE = 16000
import torch
from pprint import pprint
torch.set_num_threads(1)
NUM_PROCESS=4 # set to the number of CPU cores in the machine

Keybase proof

I hereby claim:

  • I am eschmidbauer on github.
  • I am eschmidbauer (https://keybase.io/eschmidbauer) on keybase.
  • I have a public key ASAHxHGuRHU1k5ke7a8c8KZTZCJEMpFjeBg5MceMsmrWVgo

To claim this, I am signing this object: