File size: 186 Bytes
41ba53f
 
3585493
41ba53f
3585493
1
2
3
4
5
6
import pickle

dic = pickle.load(open("./wavegru_vocoder_1024_v3_1310000.ckpt", "rb"))
del dic["optim_state_dict"]
pickle.dump(dic, open("./wavegru_vocoder_1024_v3_1310000.ckpt", "wb"))