Yehor commited on
Commit
ba703d4
·
1 Parent(s): 71a422e
export_weights.py DELETED
@@ -1,14 +0,0 @@
1
- import torch
2
-
3
- radtts_path = "models/radtts-pp-dap-model/model_dap_84000.pt"
4
- radtts_path_state = "models/radtts-pp-dap-model/model_dap_84000_state.pt"
5
-
6
- checkpoint_dict = torch.load(radtts_path, map_location="cpu")
7
-
8
- del checkpoint_dict["iteration"]
9
- del checkpoint_dict["optimizer"]
10
- del checkpoint_dict["learning_rate"]
11
-
12
- print(checkpoint_dict.keys())
13
-
14
- torch.save(checkpoint_dict, radtts_path_state)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
RADTTS-LICENSE → licenses/RADTTS-LICENSE RENAMED
File without changes
TEXT-PROCESING-LICENSE → licenses/TEXT-PROCESING-LICENSE RENAMED
File without changes