File size: 178 Bytes
587b6c9
 
 
 
 
1
2
3
4
5
6
import pickle

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