Spaces:
Running
Running
Organize
Browse files
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
|