Function main

codegeex/paddle/pt_to_pdparams.py:47–50 · view source on GitHub ↗

(opt)

Source from the content-addressed store, hash-verified

45
46
47	def main(opt):
48	state_dict = torch.load(opt.pt)
49	WalkDict(state_dict)
50	paddle.save(state_dict, opt.pdparams)
51
52
53	if __name__ == "__main__":

pt_to_pdparams.pyFile · 0.70

WalkDictFunction · 0.85

no test coverage detected