GaboChoropan commited on
Commit
1202921
·
verified ·
1 Parent(s): 82dd451

Update _run.py

Browse files
Files changed (1) hide show
  1. _run.py +1 -1
_run.py CHANGED
@@ -99,7 +99,7 @@ model = build_model(model_params, text_aligner, pitch_extractor, plbert)
99
  _ = [model[key].eval() for key in model]
100
  _ = [model[key].to(device) for key in model]
101
 
102
- params_whole = torch.load("Models/LibriTTS/epochs_2nd_00020.pth", map_location='cpu')
103
  params = params_whole['net']
104
 
105
  for key in model:
 
99
  _ = [model[key].eval() for key in model]
100
  _ = [model[key].to(device) for key in model]
101
 
102
+ params_whole = torch.load("epoch_2nd_00099.pth", map_location='cpu')
103
  params = params_whole['net']
104
 
105
  for key in model: